进入官网网址:
选择合适自己平台的安装方式,如在miniconda的虚拟环境下安装:
使用:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
先前的版本使用 previous versions of Pytorch 查找合适的版本;
附:
miniconda和虚拟环境的安装查看这篇文章:
进入官网网址:
选择合适自己平台的安装方式,如在miniconda的虚拟环境下安装:
使用:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
先前的版本使用 previous versions of Pytorch 查找合适的版本;
附:
miniconda和虚拟环境的安装查看这篇文章: