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 条评论
登录 后参与评论

相关文章

来自专栏深度学习那些事儿

深度学习-在ubuntu16.04安装CUDA9.1-总结(问题完全解决方案)

深度学习大火,为了赶上AI的班车,许多研究生本科生们都在搞深度学习。然而深度学习环境搭建必不可少,这篇文章是我多次为实验室搭建环境所积累起来的经验总结,希望所有...

5475
来自专栏木头编程 - moTzxx

ThinkCMF 初次见面/折腾一下

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

1542
来自专栏腾讯移动品质中心TMQ的专栏

【 Android 场景化性能测试】启动速度篇

传统测试启动速度的方法是录屏分帧,但这个方法耗时耗力,效率低下,于是我们使用了更为高效方便的自动化方法来进行启动速度的测试。

3851
来自专栏码神联盟

碎片化 | 第四阶段-55-OpenSessionInViewFilter组件配置解决session问题-视频

如清晰度低,可转PC网页观看高清版本: http://v.qq.com/x/page/s05686to2z4.html OpenSessionInViewFi...

33111
来自专栏企鹅号快讯

Ubuntu更新内核和NVIDIA:修复Meltdown与Spectre漏洞

正如承诺的那样,Canonical刚刚发布了新内核和Nvidia更新,以解决所有受支持的Ubuntu Linux版本中的Meltdown和Spectre安全漏洞...

2536
来自专栏lestat's blog

使用apache2反向代理访问google

引言 刚刚在v2ex上看到一篇用nginx做反向代理访问google的帖子,出于好奇,试了试用apache进行反向代理访问google,果然很好玩! 准备...

4794
来自专栏lestat's blog

开始在ubuntu下工作

家里的PC机有几个月没用了,昨天刚开机,发现SSD识别不了,经过排查,确定是盘坏了,然后只能重装

1571
来自专栏琯琯博客

awesome-sysadmin-cn资源

系统管理员 资源列表,内容包括:备份/克隆软件、云计算/云存储、协作软件、配置管理、日志管理、监控、项目管理 备份 备份软件 Amanda:客户端-服务器模型备...

52712
来自专栏数据和云

故障恢复:从未知错误ORA-600 [3712]以猜测推理达成恢复

李真旭 Oracle ACE,云和恩墨技术专家 个人博客:www.killdb.com 在Oracle数据库的日常维护中,我们可能经常会遇到一些从未见过的错误,...

2886
来自专栏iOS 开发杂谈

iOS直播app之Mac搭建nginx+rtmp服务器

前言 今年移动直播行业的兴起,很多公司都在做直播,转眼间过了大半年了,直播的各种技术相对成熟了起来。今天我们公司也追赶潮流,索性我也转战直播,下面从第一步开始...

3224

扫码关注云+社区