1.安装anaconda

见【CV】Anaconda 安装教程-CSDN博客

2.更新显卡驱动，查询CUDA版本

如果很久没有更新显卡驱动，建议更新。
在cmd中输入nvidia-smi即可查询到显卡信息如下，其中就有CUDA版本

3.使用清华镜像

1.生成.condarc文件

在anaconda prompt中输入conda config --set show_channel_urls yes
然后可以在 C:\Users\xxx 中找到.condarc文件
用记事本打开.condarc文件，将其中的内容替换为以下内容并保存

  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

在anaconda prompt中输入conda clean -i命令清除缓存

4.创建anaconda虚拟环境

打开anaconda prompt
输入conda create -n PyTorch python=3.9，其中PyTorch是虚拟环境的名字，可以自己取；笔者用的是python3.9版本
之后会有确认步骤，输入y确认等待创建完成即可

5.安装pytorch

进入pytorch官网，下拉找到下面的界面，注意CUDA版本要选择对应自己的GPU的版本
笔者使用的GPU是RTX 3050Ti Laptop，CUDA版本是11.6，之前安装了CUDA10.2版本的pytorch，在pycharm中使用pytorch时出现下面的报错，（旧版本的pytorch卸载方式可以见本文末）

在这里插入图片描述

复制这里的命令conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
打开anaconda prompt，使用conda activate PyTorch命令进入刚刚创建的虚拟环境（笔者取的名字是PyTorch）
粘贴刚刚复制的命令安装即可

6.验证安装

进入虚拟环境，输入python进入python环境，输入import torch和torch.cuda.is_available()，返回true表示安装成功
在这里插入图片描述

n.卸载pytorch

1.使用conda activate PyTorch进入anaconda的虚拟环境PyTorch（笔者取的名字是PyTorch）
2.在anaconda的虚拟环境中使用命令conda uninstall pytorch卸载之前的pytorch
3.会出现询问确认与否，输入y等待卸载完成即可
在这里插入图片描述