注: 首先确保已安装好nvidia驱动和cuda,通过如下命令查看是否已安装。
nvidia-smi nvcc -V 1、安装Anaconda清华源镜像下载地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
下载安装包进行安装
镜像下载地址: https://download.pytorch.org/whl/torch_stable.html
分别下载torch和torchvision
开始安装torch和torchvision
注: 安装过程中如果出现pip需要升级,则可以如下操作:
pip install --upgrade pip -i https://pypi.doubanio.com/simple 3、验证及再次激活PyTorch source activate pytorchpython3 #进入python环境import torch as tprint(t.cuda.is_available()) #若输出为True则说明安装成功