深度学习pytorch安装

安装CUDA

1、CUDA的版本和pytorch的版本需要相对应。具体的对应关系见Previous PyTorch Versions | PyTorch

• CUDA网址：CUDA Toolkit Archive | NVIDIA Developer
  如何知道显卡支持的CUDA版本呢？
  win+r -> cmd输入 nvidia -smi # 右上角显示支持的最大CUDA版本

官网下载对应版本的CUDA后，双击进入NVIDIA安装程序，一路默认下一步。
下载安装后，查看CUDA版本

nvcc -V

• cuDNN网址：cuDNN Archive | NVIDIA Developer
  注意：刚登陆cuDNN网址的时候，需要注册登录NVIDIA的账号。登陆之后，下载对应CUDA版本的cuDNN。
  将CUDNN拷贝到CUDA文件下。
  
  在Path中配置cuDNN的环境
  

2、安装pytorch

2.1 下载Miniconda

Miniconda网站：Miniconda — conda documentation
下载对应Python版本的，双击安装，一路默认即可。

2.2 配置虚拟环境

• 打开Anaconda Prompt命令行
  虚拟环境的好处：方便你用不同版本的python打开不同版本类型的库，如：python3.7下的TensorFlow、python3.8下的pytorch
  新建虚拟环境，输入以下代码，注意python后面输入相应的版本，如3.7，3.8等。
  
  安装完成后，进入pytorch环境
  

2.3 安装pytorch

pytorch网站：Previous PyTorch Versions | PyTorch
找到对应版本的安装命令

在pytorch虚拟环境中，安装pytorch
将上面图片中的安装命令复制进命令行即可。
安装完成后，检查是否安装成功，图示为安装成功。