New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

无法用GPU训练 #6

Open

wyfNB opened this issue Jul 23, 2020 · 1 comment

wyfNB commented Jul 23, 2020

加载完GPU后程序直接弹出，不能训练数据。有没有高版本TF的代码。

YuenSzeHong commented Jun 26, 2021 •

edited

Loading

在requirement.txt修改，

tensorflow >= 1.11.0   # CPU Version of TensorFlow.
# tensorflow-gpu  >= 1.11.0  # GPU version of TensorFlow.
pandas

把注解掉的撤销注解（把#删掉），然后把没有GPU那个注解掉即可

# tensorflow >= 1.11.0   # CPU Version of TensorFlow.
tensorflow-gpu  == 1.15.0  # GPU version of TensorFlow.
pandas

比较新版的tensorflow会出问题（2.几）

YuenSzeHong mentioned this issue

训练的时候没有用上gpu #2

Open

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment