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

PyTorch 专栏收录该内容
2 篇文章 0 订阅

安装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

我的博客即将同步至腾讯云+社区,邀请大家一同入驻。

  • 21
    点赞
  • 8
    评论
  • 92
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值