[Linux] CUDA安装(四)---- Pytorch 安装

1. 安装背景

本机的是Ubuntu 18.04版本, 同时装了python 2.7 以及python3.6. 在CUDA系列中,已经介绍安装的是CUDA10.0. 因此, 本机安装 CUDA 10.0对应pytorch 版本.

2. pytorch 官网

PyTorch https://pytorch.org/

可以看到,直接进入的 页面, 都是最新版,,所以进入previous version of pytorch

 

 

 可以看到是对应的pytorch 对应版本. 但是本机为安装Anaconda, 所有直接使用conda命令不行. 

本机使用 pip 安装,, 点击 Wheel

 

打开终端terminal

 

pip install torch==1.2.0 torchvision==0.4.0

 运行上述代码,安装即可.安装结束后.在终端输入python3, 然后输入import torch.

若未报错,则安装成功(在terminal中输入python3是因为本机安装python2.7和python3.6)