ubuntu 安装 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)

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

这里使用pip安装pytorch,我试过cuda安装,和电脑的配置没兼容,没有安装成功,后来发现使用pip安装很简单方便,就是用pip安装
首先进入pytorch官网,往下拉会看到不同的安装选项
根据自己电脑的配置选择不同的选项,例如我选择的stable版本、Linux系统、pip安装、python版本为2.7、cuda为8.0,如下图所示:
在这里插入图片描述
选择完之后下面会出现安装所需的命令,如下图
在这里插入图片描述
先后运行前两个命令就可以,网速慢的话只能慢慢等了,安装成功第一条命令后是这样的
在这里插入图片描述
安装成功第二条命令后是这样的
在这里插入图片描述
下面测试是否安装成功
命令行中输入

python
import torch
torch.__version__

没有问题的话会输出版本为0.4.1,如下图所示
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/majinlei121/article/details/83281117