Tensorflow调用GPU失败原因查找

问题查找

打开镜像,进入python,导入tensorflow,查看提示信息
Tensorflow调用GPU失败原因查找_第1张图片
这里出现警告 cudart64_101.dll not found

解决办法

找到你安装的cuda目录,进入bin文件找到cudart64_*.dll文件,可能你的文件是cudart64_102.dll,只是你只需要把这个文件改为上面提示找不到的文件名即可。
Tensorflow调用GPU失败原因查找_第2张图片

完成

查看可用设备

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

Tensorflow调用GPU失败原因查找_第3张图片
这时我们可以看到自己的GPU信息,说明已经成功了。

你可能感兴趣的:(Tensorflow,python,开发语言,后端)