1. 查看CUDA版本
cat /usr/local/cuda/version.txt
我的是 CUDA10.0
2. 安装Pytorch
2.1 打开Pytorch网址 https://pytorch.org/
2.2 如果安装最新的Pytorch,则执行默认命令:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
我需要运行的的程序要求pytorch1.0.0版本,所以选择1.0.0版本的安装:
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
3. 安装后,终端输入 conda list, 显示如下:
或则终端输入 python,import torch,无异常,则表明安装成功!