ubuntu 安装 pytorch （ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0）

ubuntu 安装 pytorch （ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0）

 

 版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.youkuaiyun.com/majinlei121/article/details/83281117

这里使用pip安装pytorch，我试过cuda安装，和电脑的配置没兼容，没有安装成功，后来发现使用pip安装很简单方便，就是用pip安装
首先进入pytorch官网，往下拉会看到不同的安装选项
根据自己电脑的配置选择不同的选项，例如我选择的stable版本、Linux系统、pip安装、python版本为2.7、cuda为8.0，如下图所示：

选择完之后下面会出现安装所需的命令，如下图

先后运行前两个命令就可以，网速慢的话只能慢慢等了，安装成功第一条命令后是这样的

安装成功第二条命令后是这样的

下面测试是否安装成功
命令行中输入

python
import torch
torch.__version__

• 1
• 2
• 3

没有问题的话会输出版本为0.4.1，如下图所示