windows下tensorflow-gpu2.5+cuda11.2+cudnn8.1环境配置

1.下载visualStidio2019(cudn要用到)

要下载2019版本,2022好像不很适配,cudn11.2以上的版本或许可以尝试

visualstidio官网是2022版本,找2019的下载Visual Studio 2019 版本 16.11 发行说明 | Microsoft Learn

勾选桌面c++开发

2.下载NVIDIA GeForce Experience(用于更新显卡驱动)

在英伟达官网即可下载下载 GeForce Experience | NVIDIA | GeForce

安装后如下,更新驱动即可

3.查看tensorflow对应的cuda及cudnn版本

在 Windows 环境中从源代码构建  |  TensorFlow (google.cn)

4.查看驱动支持cuda最高版本

下载的cuda版本需要小于等于这个版本

5.安装cuda

除了cuda的勾全部去掉,版本不同会有问题,只安装cuda即可

过程中的安装地址要记住,建议新手选择默认,后面的环境变量及cudnn直接按照地址复制就好

6.安装cudnn

是一个压缩包,解压后将其对应的目录下dll文件复制到cuda对应目录下即可

默认的cuda目录如下:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

7.环境变量添加及验证

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\cudnn\bin

将这两行添加到path里,上移至cuda后面

win+r,cmd中输入nvcc -V即可

在python终端

import tensorflow as tf
tf.test.is_gpu_available()

结果为true即安装成功

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值