conda使用國內源安裝pytorch

1、添加國內源

用記事本打開C:\Users\LZL\.condarc，添加以下內容：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
ssl_verify: true

注：沒有此文件就創建一個.condarc，路徑中的LZL需要改成你的用戶名。

2、查看本機CUDA版本

命令行輸入：nvidia-smi。如下圖所示，我的CUDA版本是11.2。

 

3、conda安裝pytorch

pytorch官網查看安裝命令，用國內源一定要去掉后面的-c pytorch，因為這是指定用官方源安裝的命令。

Linux直接在命令行，Windows在Anaconda Prompt命令行輸入：

conda install pytorch torchvision torchaudio cudatoolkit=10.2

注：cudatoolkit版本，只要不高於本機CUDA版本（11.2）都可安裝。

4、安裝完畢，查看pytorch版本