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
注意点
- 要将 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号
- tensorflow 官网有个 tensorflow版本和cuda版本对照表,大致看一下就好,只是个参考,比如我现在这个配置,那里面是没有的,它提到 tf2.6.0 对应 cuda 11.2,但实际上11.4也是可以的
- 卸载之前安装过的cuda toolkit和cuda driver时一定要卸载干净,在应用卸载界面把有nvidia字样的一律全部卸载,不然会造成版本混淆
参考资料
- https://blog.csdn.net/weixin_43760844/article/details/115612970
- https://cdmana.com/2021/12/202112111843045915.html