红联Linux门户
Linux帮助

ubuntu16.04下对theano进行cuda配置

发布时间:2016-08-25 10:13:31来源:linux网站作者:xiholix
由于cuda7.5还没有支持ubuntu16.04,如果在ubuntu16.04上安装cuda7.5会出现各种问题,处理起来比较麻烦。一个简单的解决方法是安装cuda8.0。cuda8.0下载地址(https://developer.nvidia.com/cuda-release-candidate-download)。cuda8.0目前需要先注册然后才能下载。在如下页面中我们选择runfile下载。
ubuntu16.04下对theano进行cuda配置
 
然后到该.sh文件所在的目录执行sudo sh cuda_8.0.27_linux.run
安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部:
即通过export命令 如export PATH=/usr/local/cuda-8.0/bin:$PATH
以及export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
然后使用source ~/.bashrc使该设置生效。
 
在安装时我出现了一个Failed to compile cuda_ndarray.cu: libcublas.so.8.0: cannot open shared object file
解决的办法是:sudo ldconfig /usr/local/cuda-8.0/lib64
 
本文永久更新地址:http://www.linuxdiyf.com/linux/23603.html