linux+cuda11.4安装Pytorch-GPU,一系列报错且安装了CPU版本
一、安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.4 -c pytorch -c nvidia
使用官方源速度慢且可能出现找不到包
PackagesNotFoundError: The following packages are not available from current channels:
- cudatoolkit=11.4
二、换源
方法一
用add命令添加源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
方法二
直接修改.condarc文件
channels:
- defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_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/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- 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
- 注意 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r 经常404,必要时可以删掉
- 由于更新过快难以同步,清华源不同步pytorch-nightly, pytorch-nightly-cpu, ignite-nightly这三个包
- 运行 conda clean -i 清除索引缓存
custom_channels中的源使用时注意要加 -c pytorch,pytorch才会替换
pytorch/linux-64::pytorch-1.13.0-py3.7_cpu_0
但channels下的url可以不指定-c pytorch,下载包时会自动搜索
总结
方法一不用加-c pytorch;方法二必须加-c pytorch。
三、成功安装后发现是CPU版本
conda list或者conda install时发现pytorch的版本为cpu
pytorch-1.13.0 | py3.7_cpu_0 84.3 MB pytorch
pytorch-mutex-1.0 | cpu 3 KB pytorch
torchaudio-0.13.0 | py37_cpu 6.4 MB pytorch
torchvision-0.14.0 | py37_cpu 25.2 MB pytorch
原因如下
未正确使用源或源中根本没有满足 python3.7 + cudatoolkit11.4 + pytorch1.13 的版本。
因此,要先在源下找到并指定好版本下载
果然在 https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/linux-64/ (清华源有时候没有新版本会自动到北京外国语源中搜索)中没找到pytorch-1.13.0-py3.7_cuda11.7使用
conda install pytorch=1.12 torchvision torchaudio cudatoolkit=11.3 -c pytorch