Ubuntu 16.04 LTS 安装Theano配置GPU

首先列示我的电脑使用软硬件配置:Ubuntu 16.04 LTS 64位系统,NVIDIA Geforce 840M,Anaconda python 3.5,Cuda7.5。

在详述之前,首先记录一下之前安装经历的两个坑:

2. 这一条就是使用Ubuntu 16.04 LTS安装Cuda所存在的问题。按照NVIDIA官网上提供的方法sudo dpkg -i cuda*.deb, sudo apt-get update, sudo apt-get install cuda,在apt-get update这一步,出现了Release无法下载,后面也就无法进行了,尝试了几个方法,也未能解决。后来尝试了其他方法安装Cuda,配置GPU成功,才发现,在Ubuntu下配置Theano确实容易,只是由于系统和软件的兼容性问题,导致这么多幺蛾子。最后我又把系统删除,重新安装系统及theano,确保下面所述方法无误。

下面正式介绍在Ubuntu 16.04 LTS系统下安装Theano及配置GPU的过程。

1. 更新NVIDIA显卡驱动

安装好系统之后首先在系统更新管理器中更新显卡驱动,如下图

Ubuntu 16.04 LTS 安装Theano配置GPU

点击Apply Changes

2. 安装Anaconda

在Anaconda官网选择合适的安装包下载之后,运行如下命令
  bash Anaconda3-4.1.1-MacOSX-x86_64.sh
安装好之后重启电脑,本应该在第一步重启电脑,由于我这两步是同时进行的(因为第一步更新时间较长),所以在这里重启电脑,但是重启电脑之后惊奇发现,Anaconda的gcc版本自动降为4.4,正好解决了Cuda不支持5.0以上gcc版本的问题,不需要再进行gcc降级了。

Ubuntu 16.04 LTS 安装Theano配置GPU

但是只查看gcc –version的话,其版本依然是5.4。

Ubuntu 16.04 LTS 安装Theano配置GPU

此处不知为何原因,但是后面可以成功配置GPU。

3. 安装Cuda7.5

直接 sudo apt-get install nvidia-cuda-toolkit 即可安装Cuda7.5,安装完成之后可以通过nvcc -V查看Cuda版本,

Ubuntu 16.04 LTS 安装Theano配置GPU

而且这种安装的方便之处在于不需要设置Cuda的环境变量,据知乎上吕俊奇回答“因为安装在/usr/lib/x86_64-linux-gnu/中,没有新建文件夹所以不用添加环境变量”。

4. 安装theano

Anaconda安装theano可用conda直接安装
  conda install theano

5. 配置.theanorc

生成文件 sudo gedit ~/.theanorc(注意不要漏掉theano前面的一个点)并复制如下内容,然后保存,其中cuda一项的内容为cuda所安装的位置。
  [global]
  floatX=float32
  device=gpu
  [cuda]
  root=/usr/lib/nvidia-cuda-toolkit
  [nvcc]
  flags=-D_FORCE_INLINES

至此theano的安装及配置已经完成,在python里输入import theano,显示使用的GPU设备表示GPU配置成功。

Ubuntu 16.04 LTS 安装Theano配置GPU

也可以采用Using the GPU中的测试程序对GPU进行测试。

相关推荐