Win10+CUDA11.0+cuDDNv8.0.5+RTX 3060Ti GPU深度学习主机+pytroch环境
- Win10+CUDA11.0+cuDDNv8.0.5+RTX 3060Ti GPU深度学习主机+pytroch环境
-
-
- 1.下载CUDA
- 2.下载cuDNN
- 3.安装CUDA个cuDNN
- 4.安装pytorch（方法一：手动安装）
- 5.安装pytorch（方法二：镜像源+VPN（全局模式））
- 6.测试是否安装成功
-
Win10+CUDA11.0+cuDDNv8.0.5+RTX 3060Ti GPU深度学习主机+pytroch环境
电脑系统：windows10
python版本：3.8
pytorch版本：1.7.1
CUDA版本：11.0
cuDNN版本：8.0.5适用于CUDA11.0
1.下载CUDA
https://developer.nvidia.com/cuda-toolkit-archive
选择需要的版本，这里选了11.0版本，因为3060ti只支持11.0以上版本，而pytorch当前最高支持11.0版本。
选择WIN10、本地安装，再点击下载安装包即可
2.下载cuDNN
https://developer.nvidia.com/rdp/cudnn-archive
下载cuDNN需要注册NVIDIA账号并登录，注册时安装官网提示填写用户名、密码、邮箱等，其中需要邮箱验证。
3.安装CUDA个cuDNN
打开CUDA安装包，点击自定义，勾选掉Visual Studio Intergration
安装CUDA完毕后，解压cuDNN压缩包，复制以下所有内容。
打开路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
粘贴到里面
4.安装pytorch（方法一：手动安装）
https://download.pytorch.org/whl/torch_stable.html
找到版本cu110/torch-1.7.1%2Bcu110-cp38-cp38-win-amd64.whl
cu110/torchvision-0.8.2%2Bcu110-cp38-cp38-win-amd64.whl
5.安装pytorch（方法二：镜像源+VPN（全局模式））
我用的方式二反正，其实下载的东西是一样的。
以管理员身份打开命令窗口
激活conda创建的虚拟环境my_pytorch
然后进入下方链接，选择合适的配置，生成安装代码
https://pytorch.org/get-started/locally/
先添加清华镜像源，打开全局模式的VPN（不开也行，就是慢）
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c nvidia
6.测试是否安装成功
python
import torch
import torhvision
torch.__version__
