一、安装途径推荐
安装和验证的方法就不讲的,其他博客写的很详细。
主要说一下安装过程遇到的问题:
有条件的建议推荐还是搭个梯子,用官方源速度也很快(原因后面说)
1.安装gpu版时先看好NVIDIA驱动的版本:
桌面鼠标右键NVIDIA控制面板→左下角系统信息→组件
根据CUDA版本在这里找对应的pytorch、torchvision版本:
https://download.pytorch.org/whl/torch_stable.html
然后wins或linux在anaconda prompt下使用与下面例子相同格式命令:
conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=11.0 -c pytorch
其他系统参考:https://pytorch.org/get-started/previous-versions/
torch.cuda.is_available()结果为false
经过反复试错要点总结如下:
使用GPU的同学们
1.最好用官方源,实在没有梯子下载速度慢,其他源也行,主要是源的文件要齐全,不要有缺失;
2.一定要注意必须根据自己的NVIDIA的CUDA驱动版本在上面网址中找好对应的pytorch版本号和torchvision版本号,做到这点,大概率为true。…
windows下查看显卡编号和使用情况
nvidia-smi