目录
写在前面(两个深坑):
1.一开始尝试安装cuda10.1,cudnn7.6.5和tensorflow_gpu2.3.0,整个安装都没问题,但是测试的时候一直找不到GPU(tf.test.is_built_with_gpu_support()=false)。最后猜测是版本问题。虽然cuda10.1支持2.3.0,但很容易安装不上,因此最好还是安装tensorflow_gpu2.2.0版本。
2.如果显卡是NVIDIA RTX 30X0系列的话,建议直接安装cuda11及对应版本。如果装cuda10的话可能会导致动态库载入很慢并且网络训练结果一直是NAN。
3.安装期间没遇到过报error的问题,都是上面这种不报错但是查不出来的问题。
正文
1.检查需求
查看显卡的算力版本是否满足需求,确认要安装的各个版本间的对应关系。
1.GPU:NVIDIA Geforce GTX 1050 Ti
查询方式:(1)桌面直接右键-NVIDIA控制面板-左下角 系统信息
(2)此电脑右键-管理-设备管理器-显示适配器
2.GPU算力:6.1
查询官方链接:https://developer.nvidia.com/zh-cn/cuda-gpus#compute
3.支持最高版本CUDA:
查询方式:桌面直接右键-NVIDIA控制面板-左下角 系统信息-组件
我的显示最高11.3,也就是说cuda版本不能装超过11.3的。
4.cuda、cudnn、tensorflow-gpu之间的版本对应关系:
官方链接:https://tensorflow.google.cn/install/source_windows#gpu