如果通过传统的pip方式安装,则必装出CPU的版本,如果想直接使用系统的默认配置的Pytorch+GPU方案,只需要两个步骤即可

第一,卸载已经在环境下安装的包

conda activate 你的环境名
pip uninstall torch torchvision -y

第二,指向系统原生的设置

echo "import sys; sys.path.append('/usr/local/lib/python3.10/dist-packages')" > ~/anaconda3/envs/你的环境名/lib/python3.10/site-packages/system_torch.pth

记得在执行命令时替换自己的环境名

然后在环境中再进行一次调用的验证

python -c "import torch; print(torch.__version__, torch.cuda.is_available())"

看到命令返回True即为成功

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐