pip安装参考
pip install --trusted-host download.pytorch.org torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
conda 安装
先设置镜像源,如清华的conda镜像(可参考其他博客)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
官方安装的命令是(版本1.4 GPU):
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
但要用国内源,我发现不能用-c这一段,直接用
conda install pytorch torchvision cudatoolkit=10.1