Pytorch安装-Ubuntu 安装 Pytorch教程

个人微信公众号：AI研习图书馆

ID：（Art-Intelligence）
欢迎关注，交流学习，共同进步~

1. 引言

关于显卡驱动安装，cudn安装，cudnn安装，可以参看我的另一篇博客，caffe安装的前半部分，有详细安装过程：
https://blog.csdn.net/weixin_44222014/article/details/103130053

本文主要介绍pytorch安装。

本机环境：ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0+cudn5

2. 安装方法

2.1 安装

首先进入pytorch官网，往下拉会看到不同的安装选项
根据自己电脑的配置选择不同的选项，例如我选择的stable版本、Linux系统、pip安装、python版本为2.7、cuda为8.0，如下图所示：
pytorch官网

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

先后运行前两个命令就可以，网速慢的话只能慢慢等了，也可以离线下载，然后进行安装。

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

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

2.2 测试

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

python
import torch
torch.__version__

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

以上只是基于pytorch0.4.1的安装方法，举一反三，同理可得，其他版本也是如此流程安装，比较简单，建议离线下载，然后安装~

祝你成功，深度学习大有可为~

您的支持，是我不断创作的最大动力~

欢迎点赞，关注，留言交流~

深度学习，乐此不疲~

欢迎关注个人微信公众号~