1.主要是显卡驱动的版本，尽量更新到最新版（更新到最新就不要考虑驱动版本和cuda版本的对应，会方便很多）
驱动版本查看：输入命令nvidia-smi，如图
如果出现以下提示：
’nvidia-smi’ 不是内部或外部命令，也不是可运行的程序
参考这里https://blog.csdn.net/shuiyixin/article/details/99897640
显卡驱动更新教程：https://zhidao.baidu.com/question/1951584892469977108.html
显卡驱动与cuda的版本对应图：
2.需要安装的包主要包括三个：pytorch，cudatoolkit，cudann（这三个包同样要注意版本对应） torchvision和torchaudio按需安装。
清华镜像设置：进入C:\Users\Administrator，找到.condarc文件，粘贴我下面的代码覆盖原来的
ssl_verify: true
show_channel_urls: true
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
下载三个主要包的命令：
conda install pytorch=你所需版本号
conda install cudatoolkit=你所需版本号
conda install cudann=你所需版本号
cudatoolkit与pytorch的版本对应：可以看pytorch的版本号，例如pytorch=1.7.1+cu101代表对应cudatoolk10.1版本的pytorch1.7.1。有点拗口，懂就行。
cudatoolkit与cudann的版本对应：在安装完pytorch和cudatoolkit后可以输入命令conda search cudnn --info有版本的对应信息，如下图：
最后附上pytorch官网：https://pytorch.org/
搞了一天，最后原来是更新个显卡驱动这么简单的事。。。