Ubuntu16.04 + CUDA10.0 + Pytorch1.0.0 安装

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，无异常，则表明安装成功！