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号查询对应参考下面链接：
   • NVIDIA 驱动程序下载 这个要选择一下上面的条件，然后点击搜索后就能看到对应version
   • CUDA Toolkit and Corresponding Driver Versions (Table 3)
   • cuDNN Archive 下载cuda toolkit对应版本，文件中有 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