让tensorflow在GPU上运行

Question

我一直在努力让这个工作永远，仍然没有运气

我:

• GTX 1050 Ti(联想军团笔记本电脑)
• 笔记本电脑也有一个英特尔UHD显卡630(我不确定如果这是干扰?)
• 蟒蛇
Visual Studio
Python 3.9.13
• CUDA 11.2
• cuDNN 8.1
• 我将这些添加到PATH中:C:Program FilesNVIDIA GPU计算工具包CUDAv11.2binC:Program FilesNVIDIA GPU计算工具包CUDAv11.2libnvvp
最后我安装了tensorflow并创建了自己的环境

我仍然不能让它读取我的GPU

基本遵循https://www.youtube.com/watch?v=hHWkvEcDBO0&t=295s

我还是没有运气。

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

只产生关于CPU的信息

有人能帮忙吗?

Answer 1

可以将tensorflow升级到2.0。这应该能解决你的问题。

Answer 2

检查您的tensorflow版本和与GPU的兼容性，更新您的GPU驱动程序。CUDA 9/10就可以了。

遵循官方tensorflow链接:https://www.tensorflow.org/install/pip windows-native_1

在anaconda的相同环境中完成所有步骤。