前提:已经安装anaconda
pytorch官网:查找对应的版本和安装指
PyTorchhttps://pytorch.org/
进入anaconda base环境
conda activate base
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装成功显示:
在刚刚安装PyTorch的conda环境中要使用Jupyter notebook需要单独安装,安装指令如下:
conda install nb_conda -y
新增:检查torch是否安装成功: