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

相关文章

来自专栏人工智能LeadAI

TensorFlow从0到1 | 第十八章: 升级手记:TensorFlow 1.3.0

《TensorFlow从0到1》写到现在,TensorFlow的版本也从当时的1.1.0迭代到了8月初发布的1.3.0。可以预见在未来很长一段时间里,它仍会持续...

2737
来自专栏磐创AI技术团队的专栏

Ubuntu 安装 tensorflow-gpu 1.4 +CUDA 8.0 +cuDNN详细教程

1642
来自专栏软件开发 -- 分享 互助 成长

win10使用u盘装回win7

背景:一朋友要我帮忙把系统从win10装回到win7,因为做IT的嘛,想想也难不倒我,况且以前也经常重装系统,硬盘里就有win7的系统,于是很爽快的答应了。电脑...

1809
来自专栏机器之心

教程 | 如何使用 Kubernetes 轻松部署深度学习模型

本文展示了如何用 Keras 构建深度学习模型的简单示例,将其作为一个用 Flask 实现的 REST API,并使用 Docker 和 Kubernetes ...

791
来自专栏PaddlePaddle

【PaddlePaddle】Windows版快速安装包正式上线!

正在钻研深度学习,学习PaddlePaddle,或者是正在用Windods系统的小伙伴们,再也不用特意装Linux系统,借助快速安装包就可以下载PaddlePa...

1074
来自专栏机器学习算法全栈工程师

Windows下安装Scikit-Learn

0. 引言 自从机器学习大火起来以后,做机器学习最热门的语言应该说是非Python莫属,原因大致有以下几个方面:1. Python语言简单易学,语法简单灵活;2...

3426
来自专栏张戈的专栏

WP-PostViews Plus统计插件在TwentyTen主题下的使用

为了博客的文章浏览统计,让我这个不怎么熟悉 PHP 的菜鸟硬是从昨晚折腾到现在! 主要原因是我不太熟悉 PHP 的结构,结果被网上的教程搞得云里雾里,通过度娘搜...

28610
来自专栏Python中文社区

手把手教你用1行代码实现人脸识别

專 欄 ❈Kangvcar,Python爱好者,简书活跃作者,欢迎关注,打赏支持。❈ 环境要求: Ubuntu17.10 Python 2.7.14 环境...

1.2K10
来自专栏Deep learning进阶路

caffe随记(九)---利用FCN和已有的model进行图像语义分割

1、下载caffemodel 本例中我们使用的是pascalcontext-fcn32的模型,这个下载链接在它的文件夹里有,就是url那个文件 下载  pas...

2910
来自专栏计算机视觉life

OpenCV学习入门(二):Image Watch神器

Image Watch是在visual studio 2012及以上版本上使用的一款OpenCV工具,能够在调试过程中实时显示内存中矩阵Mat(存放图像,数组等...

1865

扫码关注云+社区