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官网

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