确认版本

1. 先在终端查看cuda版本，我这里是12.6

nvcc -V

2. Pytorch要与cuda版本对应，在官网https://pytorch.org/查看对应关系，cuda12.4以后使用12.4即可，这里会把安装命令也给出来，直接用命令安装走的是官方源，速度很慢。

注意：如果已安装pytorch还出现报错 ModuleNotFoundError: No module named ‘torch‘，也用此方法确认torch版本是否正确

换国内源

1.打开终端执行如下命令，会在当前用户目录下生成一个.condarc的文件，可以用命令查看这个文件的路径

conda config --set show_channel_urls yes
conda config --show-sources

2.复制如下内容到.condarc文件中

channels:
  - 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
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

3. 清除历史索引缓存

conda clean -i

安装pytorch

此时使用pytorch官网的命令安装即可，如果打不开官网，直接复制下面这个

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia