2020-12-09-win10下GTX3060TI+CUDA+cudnn+pytorch1.7.0安装

    最近公司新配置了一台AI工作站，显卡用的刚发售的3060Ti，之前做的深度学习环境镜像中的cuda是10.1的，没法用了，就重新配置了一下环境，在这里做个记录，顺便给有需要的朋友看。

    我默认你的电脑中已经安装了GTX3060Ti的显卡驱动（没有装的可以去NVIDIA官网下载安装），并且已经配置好了anaconda（python3.7版本，网上安装教程很多）。

0. 准备安装包

链接：https://pan.baidu.com/s/1cwhvZbjEwXQJk2mIM1XPGg，提取码：idpc

cuda_11.0.2_451.48_win10：CUDA11.0版本

cudnn-11.0-windows-x64-v8.0.5.39：对应CUDA11.0版本的cudnn包

torch-1.7.0+cu110-cp37-cp37m-win_amd64：对应CUDA11.0版本的torch离线安装包

torchvision-0.8.1+cu110-cp37-cp37m-win_amd64：对应CUDA11.0版本的torch离线安装包

torchaudio-0.7.0-cp37-none-win_amd64：对应CUDA11.0版本的torch离线安装包

1. 安装CUDA环境

    1）首先安装cuda_11.0.2_451.48_win10，一路next即可；

    2）解压cudnn-11.0-windows-x64-v8.0.5.39后可以看到一个解压出来的cuda文件夹，把里面的文件夹复制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0”中即可;

    3）查看CUDA是否安装成功，cmd中输入：

nvcc -V

2. 安装新版pytorch

    2.1 卸载老版安装包（如果之前未安装torch，则跳过）

    如果之前有老版pytoch，需要先运行以下代码，把老版卸载掉。

pip uninstall torch

pip uninstall torchvision

---

2.2 离线安装torch包

    使用pip安装的朋友应该都配置了清华源，但是清华源中目前还没有torch1.7.0中相关的windows安装包，所以需要下载官方包安装，在前面已经分享，我们直接下载安装。

    2.2.1 还是使用pip安装，cmd中运行如下指令，离线安装:

pip install "你下载该文件存放路径\torch-1.7.0+cu110-cp37-cp37m-win_amd64.whl"

pip install  "你下载该文件存放路径\torchvision-0.8.1+cu110-cp37-cp37m-win_amd64.whl"

pip install  "你下载该文件存放路径\torchaudio-0.7.0-cp37-none-win_amd64.whl"

---

完，有什么不明白的可以留言讨论。