conda使用國內源安裝pytorch
阿新 • • 發佈：2021-08-20
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）都可安裝。