admin管理员组文章数量:1652351
目录
写在前面(两个深坑):
正文
1.检查需求
2.更新nvidia驱动(不一定需要)
3.安装cuda
4.安装cudnn
5.安装tensorflow-gpu
写在前面(两个深坑):
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/zh-cn/cuda-gpus#compute
3.支持最高版本CUDA:
查询方式:桌面直接右键-NVIDIA控制面板-左下角 系统信息-组件
我的显示最高11.3,也就是说cuda版本不能装超过11.3的。
4.cuda、cudnn、tensorflow-gpu之间的版本对应关系:
官方链接:https://tensorflow.google/install/source_windows#gpu
本文标签: NVIDIA
版权声明:本文标题:win10安装nvidia驱动,cuda10.1,cudnn7.6.5和tensorflow_gpu2.2.0 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/dianzi/1729588777a1207591.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论