PyTorch安装

• 第一步查看自己电脑支持的CUDA版本

  #打开cmd
  nvidia-smi
  

  可以看到CUDA Version：11.6，说明我们的显卡最高支持11.6这个版本。

• 第二步下载**CUDA**并安装（官方其他版本下载网页）

• 第三步PyTorch官网查询对应CUDA版本的安装

  界面如上图所示，但是对应我们的显卡CUDA版本并没有，所以我们需要去这个页面寻找Previous PyTorch Versions | PyTorch

  pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
  

• 第四步Anaconda Prompt安装PyTorch

  #创建一个虚拟环境,-n指定环境名称，同时将python设为3.8
  conda create -n pytorch python=3.8
  #进入虚拟环境
  conda activate pytorch
  #粘贴第三步得到的安装命令
  pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
  

• 第五步检查安装是否成功

  python -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available: {torch.cuda.is_available()}')"
  

  如果出现True说明我们安装成功