Ubuntu14下cuda-7.5、tensorflow安装(enabling cuda 3.0)
本文主要介绍在Ubuntu 14.04(64-bit)下安装支持NVIDIA GPU计算能力3.0的 tensorflow(版本0.7.1),如果计算能力在(3.5-5.2)直接按照官方教程安装。
我在百度云 上传了一份官方0.7.1的whl安装包,密码:dfmg ; 以及一份我自己编译的安装包,密码:h4y9,支持GPU计算能力3.0
1. 配置依赖环境
本文以python2为例
# Ubuntu/Linux 64-bit
$ sudo apt-get install build-essential python-pip python-dev
2. 安装Cuda-7.5 下载cuda
本文使用.run文件安装,安装cuda会出现ubuntu无法进入桌面的问题,王云飞师兄发现安装cuda时不装openGL可以解决这个问题,目前没发现不装openGL会造成问题。cuda包中自带NVIDIA显卡驱动,故不需要在此安装驱动。
- 禁止nouveau 参考官网
$ sudo gedit /usr/lib/modprobe.d/blacklist-nouveau.conf
# 在打开的编辑器中输入下面两行,保存并关闭
blacklist nouveau
options nouveau modeset=0
# Regenerate the kernel initramfs
$ sudo update-initramfs -u
- 开始安装
注意询问你安装 openGL时候,选择no,这样就不会发生装完lightdm无法启动的问题了
ctrl+alt+F1进入tty1
# 关闭lightdm
$ sudo service lightdm stop
# 进入cuda的下载目录(本文是Downloads)
$ cd Downloads/
# 更新
$ sudo apt-get update
# 安装(空格或pagedown都可往下读协议)
# 安装 openGL 时候,选择no 其他默认即可
$ sudo sh cuda_7.5.18_linux.run
# 安装完成后启动lightdm
$ sudo service lightdm start
# 启动后 ctrl+alt+F7 进入tty7
3. 环境变量设置
这里参考官方教程 6.Post-installation Actions
The
PATH
variable needs to include/usr/local/cuda-7.5/bin
TheLD_LIBRARY_PATH
variable needs to contain/usr/local/cuda-7.5/lib64
on a 64-bit system, and/usr/local/cuda-7.5/lib
on a 32-bit system
#To change the environment variables for 64-bit operating systems:
$ export PATH=/usr/local/cuda-7.5/bin: