pytorch的安装与配置//CUDA与cuDNN版本的选择

pytorch的安装与配置：

CUDA与cuDNN版本的选择

1、打开NVIDIA设置，点击“帮助”

2、进入“系统信息”，显卡及驱动版本如红框所示，驱动版本为451.67

3、根据驱动版本和CUDA的对应关系，在此确定CUDA版本为CUDA10.2（因为2020年9月pytorch可支持的最大CUDA版本为10.2）

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions

4、上CUDA官网下载CUDA10.2版本及其补丁
https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

5、参考网上的教程进行CUDA的安装
此处推荐一个教程：https://www.cnblogs.com/arxive/p/11198420.html
6、根据CUDA版本，安装cudnn
cuDNN官网：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10

7、安装pythorch
根据对应的CUdA版本安装对应的pytorch
官网：https://pytorch.org/

（anaconda prompt）进入环境，输入指令：
pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html