Win10下 anaconda3 + cuda11.3 + TensorFlow_GPU2.6.0 + Pytorch_GPU1.10.1

xiao_yun_zi

已于 2022-12-04 15:42:07 修改

阅读量1.1w

点赞数 24

分类专栏： AI 文章标签： python tensorflow windows pytorch

于 2022-04-09 20:56:29 首次发布

本文链接：https://blog.csdn.net/xiao_yun_zi/article/details/124046964

版权

本文详细介绍了在Windows 10环境下，如何安装CUDA11.3、cuDNN8.2.1，以及如何配置Anaconda来安装TensorFlow_GPU 2.6.0和Pytorch_GPU 1.10.1。通过遵循步骤，包括驱动安装、环境配置、库安装和验证，确保GPU支持的深度学习框架成功运行。

摘要由CSDN通过智能技术生成

Win10 + CUDA11.3 + tf_GPU2.6.0 + pytorch_GPU1.10.1

软件版本
- TensorFlow_GPU
- Pytorch_GPU
第一步：安装N卡驱动
第二步：安装Anaconda
第三步：CUDA、cuDNN的安装
第四步：安装TensorFlow_GPU
第五步：安装Pytorch_GPU
相关参考

软件版本

个人N卡2070 super，驱动512.15，支持最新的CUDA11.6（方法：英伟达控制面板—>帮助—>系统信息—>组件）。但TensorFlow_GPU 2.6.0、Pytorch_GPU 1.11.0目前并未适配到CUDA11.6。
在这里插入图片描述

TensorFlow_GPU

tf_ GPU 2.6.0 需搭配 CUDA11.2，cuDNN8.1，python3.6~3.9
tensorflow gpu版对CUDA、cuDNN、python的要求

Pytorch_GPU

pytorch_GPU 1.11.0 支持 CUDA10.2、CUDA11.3

点击查看torch、torchvision、torchaudio版本对应关系

CUDA	CUDAToolkit	Pytorch
11.3	11.3	1.11.0，1.10.1，1.10.0，1.9.1，1.9.0，1.8.1
11.1	11.1	1.8.0
11.0	11.0	1.7.1，1.7.0
10.2	10.2	1.11.0，1.10.1，1.10.0，1.9.1，1.9.0，1.8.1，1.8.0，1.7.1，1.7.0，1.6.0，1.5.1，1.5.0
10.1	10.1	1.7.1，1.7.0，1.6.0，1.5.1，1.5.0，1.4.0
10.0	10.0	1.2.0，1.1.0，1.0.1，1.0.0
9.2	9.2	1.7.1，1.7.0，1.6.0，1.5.1，1.5.0，1.4.0，1.2.0
9.0	9.0	1.1.0，1.0.1，1.0.0
8.0	8.0	1.0.0

尝试了 tf_GPU=2.6.0、CUDA=11.2、cuDNN=8.2.0、pytorch_GPU=1.8.0 or 1.10.1，结果 tf 没问题，pytorch总失败。最后安装 tf_GPU=2.6.0、CUDA=11.3、cuDNN=8.2.1、pytorch_GPU=1.10.1 两个都成功。