Theano安装笔记

由于实验需要,近三个月来,安装过十几次Theano,基本上每次都是从最基本的nvidia driver装起。总结一些粗浅的安装心得。

GPU:Nvidia K40, M40, M60

软件环境:Unbuntu 14 以及Unbuntu 16

(一)检测已有的环境

用Nividia-smi查看是否已经安装nvidia adapter graphic driver

用nvcc -V查看cuda是否已经装好,若要详细检查,还需要对cuda-sample进行make,然后./deviceQuery

(二)值得说明的一些点

1.nvidia driver(下载:)与cuda(下载:https://developer.nvidia.com/cuda-downloads,官方安装说明:http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-installation)里面的driver是一回事,如果先用nvidia driver装好了,就在cuda安装的时候直接从cuda toolkits装起。

2.Unbuntu16中,在用nvidia driver安装完成后,再在cuda安装时候从driver装起,安装过程不报错,但原来的driver会损坏掉,还得需要再重新装一遍

3.感觉最好的办法是直接用cuda安装从driver装起,这种一站式的方式更便捷,但是按cuda的官方安装说明(http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-installation)来总是有点别扭,因为它总是让你预检查device node是否存在。可是你不装driver,/dev/nvidia*怎么可能存在。

(三)安装步骤:

1.安装cuda或nvidia driver+cuda toolkits

2.设置PATH以及LD_LIBRARY_PATH(export in /etc/profile or ~/.bashrc),编译cuda-samples,运行./deviceQuery,查看Cuda是否安装好。

至此,可以通过nvcc -V查看cuda版本号,可以nvidia-smi查看gpu的使用状态

cat /proc/driver/nvidia/version查看driver版本号

3.安装minconda(安装新软件方便)或anaconda(软件全),目前最好用2.7版本的

4.采用conda install安装theano的各种依赖包

5.conda install theano pygpu

6.新建/home/username/.theanorc,配置theano,cuda_root要写进去

[cuda]

root=/usr/local/cuda-8.0/bin

6.安装cudnn:将头文件.h以及lib*文件copy到对应的cuda目录下,在cuda lib64目录中建立软连接

有些时候必须强制设置THEANO_FLAGS=dnn.enabled=True时,cudnn才生效,不知为啥?

注意:有些地方需要重启生效

原文地址:https://www.cnblogs.com/shuzirank/p/7293861.html