win10安装nvidia驱动,cuda10.1,cudnn7.6.5和tensorflow_gpu2.2.0

目录

写在前面(两个深坑):

正文

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.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

  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值