我的电脑已经安装了cuda和cudnn,也安装了tensorflow-gpu,但是跑的时候无法调用GPU运算
通过一下代码检测GPU是否被使用:
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
print(device_lib.list_local_devices())
output
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
但是我这里只有一个CPU被调用, GPU无法调用。
下面介绍解决方法
注意:最重要的问题,你的cuda版本要配置对应的tensorflow-gpu。
查看你的cuda对应tensorflow-gpu版本:https://tensorflow.google.cn/install/source_windows
我的cuda版本为11.0, 在命令行(CMD)中输入nvidia-smi查看。最终发现自己的tensorflow-gpu版本装错了,装的系统默认的最新版本2.6.0,而我应该装2.4.0,怪不得报这样的错。
装成功后的检测结果界面是这样的:
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 14584989243831407567
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3127299278
locality {
bus_id: 1
links {
}
}
incarnation: 9780177882687983642
physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"
安利一下大家需要的版本:
Windows最新的版本如下:
Linus/Macos最新的版本如下: