进入官网网址:

PyTorch

选择合适自己平台的安装方式,如在miniconda的虚拟环境下安装:

  使用:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

先前的版本使用 previous versions of Pytorch 查找合适的版本;

附:

miniconda和虚拟环境的安装查看这篇文章:

ubuntu16.04 安装miniconda_猫猫与橙子的博客-CSDN博客

更多推荐