Ubuntu16.04环境下PyTorch简易安装教程

安装NVIDIA GPU显卡驱动

如果需要安装cuda版本的PyTorch,电脑也有独立显卡的时候,一般需要更新一下Ubuntu独立显卡驱动。否则即使安装了cuda版本的PyTorch也没办法使用GPU。

因为我的笔记本电脑显卡是NVIDIA的,所以首先去官网:NVIDIA显卡驱动下载,查看适合自己显卡的驱动,下载runfile文件:NVIDIA-Linux-x86_64-384.98.run。

然后按Ctrl + Alt + F1到控制台,关闭当前图形环境:

sudo service lightdm stop

卸载可能存在的旧版本NVIDIA 驱动(这一步是必须的,不然重启会有问题):

sudo apt-get remove --purge nvidia*

安装驱动可能需要的依赖:

sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic

把 nouveau 驱动加入黑名单并禁用用 nouveau 内核模块(如果新重装的系统,没有驱动起作用,忽略):

sudo nano /etc/modprobe.d/blacklist-nouveau.conf

在文件 blacklist-nouveau.conf 中加入如下内容:

blacklist nouveau options nouveau modeset=0

保存退出,执行:

sudo update-initramfs -u

然后重启:

reboot

重启后再次进入字符终端界面(Ctrl + Alt + F1),并关闭图形界面:

sudo service lightdm stop

进入之前NVIDIA驱动文件下载目录,安装驱动:

sudo chmod u+x NVIDIA-Linux-x86_64-384.98.run
sudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files

-no-opengl-files表示只安装驱动文件,不安装OpenGL文件。这个参数不可忽略,否则会导致登录界面死循环。

最后重新启动图形环境:

sudo service lightdm start

可以通过以下命令确认驱动是否正确安装:

cat /proc/driver/nvidia/version

值得注意的是,安装完显卡驱动后,桌面可能会出现菜单任务栏及窗体最大化最小化消失问题。解决办法如下:

首先安装compiz配置管理器:

sudo apt-get install compizconfig-settings-manager

然后,打开配置管理器:

sudo ccsm

并在”general”启用OpenGL,在”Desktop”里启用Ubuntu Unity Plugin,然后返回桌面。

接着,重置compiz:

dconf reset -f /org/compiz/

最后注销重启:

gnome-session-quit

至此,NVIDIA显卡驱动安装成功。

科学上网的简易方法

进入PyTorch官网: PyTorch

显示如下:

许多教程说在“Run this command”下会显示安装PyTorch的命令。如果Ubuntu下已经配置过Anaconda的话,可以直接使用conda命令安装成功PyTorch:

conda install pytorch torchvision cuda80 -c soumith

其中,pytorch是软件,torchvision是pytorch使用的数据包和库,cuda80是GPU版本。

但是基于特殊的网络国情,一般无法查看“Run this command”之后的指令,安装也不会成功。

下载PyTorch

不能科学上网的话,使用另外一种方法来安装PyTorch。首先下载所需版本的PyTorch。在PyTorch主页上点击“Click here for previous versions of PyTorch”,如上图右下角所示。

找到所需版本,如上图所示,下载torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl。其中,0.2.0表示PyTorch版本,cp36表示支持Python3.6,cuda80表示支持GPU版本,电脑有GPU的话须选择该项。

安装PyTorch

进入PyTorch的下载目录,使用pip命令安装:

pip install torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

安装torchvision

安装torchvision比较简单,可直接使用pip命令安装:

pip install torchvision

至此,pytorch和torchvision已安装完毕。

更新numpy

安装成功pytorch和torchvision后,打开ipython,输入:

import torch

可能会出现报错的情况,如下所示:

ImportError: numpy.core.multiarray failed to import

这是因为numpy的版本需要更新,直接使用pip更新numpy:

pip install numpy

至此,PyTorch安装成功:

最后,验证GPU是否可用:

返回True,表明已经可以使用GPU了!

参考链接:

linux下 /boot 分区空间不足及其衍生问题

ubuntu16.04 安装NVIDIA和CUDA8.0

ubuntu安装显卡驱动后,右侧菜单及窗体最大最小化消失问题

Ubuntu 17.04 安装 pytorch and pytorchvision

Pytorch如何安装,Linux安装Pytorch,OSX安装Pytorch教程

ImportError: numpy.core.multiarray failed to import

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏吉浦迅科技

ubuntu12.04 cuda6 bumblebee安装图解

花了一些时间,尝试着在ubuntu12.04上安装cuda6.把过程记录下来,给自己和同我一样小白的人以借鉴。 1 我要做什么 作为一只cuda菜...

3166
来自专栏SnailTyan

Linux下Caffe、Docker、Tensorflow、PyTorch环境搭建(CentOS 7)

注:模型的训练、测试、部署都可以通过Docker环境完成,环境问题会更少。 1. CUDA 8.0安装 ? Config env variables # CUD...

3070
来自专栏数据科学与人工智能

【机器学习】TensorFlow安装与测试

官网:http://tensorflow.org/ 安装步骤: 1、sudo apt-get install python-pip python-dev pyt...

38510
来自专栏用户画像

MAC使用pip安装Tensorflow

官网英文版安装教程:https://www.tensorflow.org/install/install_mac#common_installation_pro...

672
来自专栏nummy

python支持tab补全与历史命令

Python 命令行默认是不支持tab补全与上下查看历史命令的,而IPython是支持这一功能的,那如何让Python支持上述两个功能呢?

943
来自专栏ytkah

怎么采集dedecms自定义内容模型

  有时我们需要用到dedecms提供的自定义内容模型功能去添加自定义内容模型来满足需求,那么dedecms自定义内同模型怎么添加采集规则呢?打开“/dede/...

2686
来自专栏用户2442861的专栏

express 与 express-generator

其实express只是一个框架,那么npm install -g express 也仅仅是安装了这个框架,其作用是自己构建express项目的时候,库已经可以引...

511
来自专栏生信宝典

如果你经常用PubMed,那么这个插件将非常好用!

1315
来自专栏marsggbo

批量下载Coursera及其他场景上的文件

以下方法同样适用于其他场景的批量下载。 最近在学习Coursera退出的深度学习课程,我希望把课程提供的作业下载下来以备以后复习,但是课程有很多文件,比如说脸部...

3405
来自专栏Youngxj

emlog评论添加表情教程

1102

扫码关注云+社区