win11+R7-5800+gtx3060 搭建 tensorflow+pytorch 环境

本机配置

  • cpu: AMD R7-5800
  • 显卡: Geforce RTX 3060
  • 操作系统:windows11

目标

  • 搭建深度学习环境,用 gpu 跑 tensorflow 和 pytorch 模型

主要配置

CUDA driver

471.41-notebook-win10-64bit-international-nsd-whql.exe

CUDA toolkit

cuda_11.4.2_471.41_win10.exe,虽然写的是 win10,但是可以在win11上用。

cudnn

cudnn-11.4-windows-x64-v8.2.4.15.zip
要注册账号登录后才可以下载,下载后解压,把三个目录的文件对应着复制到 cuda toolkit 目录中去

tensorflow,tensorflow-gpu

pip install tensorflow==2.6.0 tensorflow-gpu==2.6.0

pytorch

pip install torch==1.8.0

注意点

  1. 要将 CUDA driver 和 CUDA toolkit 的版本对其,我这里都用的 11.4 版本,更具体要看一下version号 471.41,可以通过version号把上述三者对应起来,其中三个version号查询对应参考下面链接:
  2. tensorflow 官网有个 tensorflow版本和cuda版本对照表,大致看一下就好,只是个参考,比如我现在这个配置,那里面是没有的,它提到 tf2.6.0 对应 cuda 11.2,但实际上11.4也是可以的
  3. 卸载之前安装过的cuda toolkit和cuda driver时一定要卸载干净,在应用卸载界面把有nvidia字样的一律全部卸载,不然会造成版本混淆

参考资料

  • https://blog.csdn.net/weixin_43760844/article/details/115612970
  • https://cdmana.com/2021/12/202112111843045915.html
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值