我这边的电脑环境是Ubuntu 14.04.4,显卡是GTX960M
查看TensorFlow所需要的CUDA和cuDNN对应版本。
https://tensorflow.google.cn/install/source
灵感来源于这里 http://blog.csdn.net/hanlin_tan/article/details/51235206 。不过我单独下载显卡驱动,会有很多报错,然后还是进不去桌面。
另外如果CUDA版本(这里是8.0)和CUDA自带的显卡驱动版本不匹配(自带的375.62),我在软件包管理器新立得(synaptic)里试过375.66以及375.82和当前最新的384.90全都用不了,具体表现是运行测试程序deviceQuery报错35,最后Failed。
卸载显卡驱动使用命令:sudo nvidia-installer -uninstall
查看电脑显卡使用的驱动版本可以使用命令
nvidia_smi
cat /proc/driver/nvidia/version
解决办法就是在终端模式,关闭桌面管理器lightdm,然后卸载显卡驱动,接着重新安装CUDA。卸载CUDA使用命令
sudo ./usr/local/cuda/bin/uninstall_cuda_8.0.pl
在这里看到的显卡驱动的最新版本
https://github.com/floydhub/dl-setup
https://launchpad.net/~graphics-drivers/+archive/ubuntu/ppa
1) 创建文件 /etc/modprobe.d/blacklist-nouveau.conf 添加如下内容,
blacklist nouveau
options nouveau modeset=0 // 关闭系统自带的显卡驱动nouveau
2) 重新生成 kernel initramfs
$ sudo update-initramfs -u
进入系统后使用命令 lsmod | grep nouveau查看,若显示为空则禁用成功
3) 重启系统,Ctrl + Alt + F1 进入命令行模式,执行,
$ sudo service lightdm stop // 关闭桌面服务
$ wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run // 获取CUDA程序
$ wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/patches/2/cuda_8.0.61.2_linux-run // 获取补丁程序
$ sudo a+x cuda_8.0.61_375.26_linux.run
$ sudo sh cuda_8.0.61_375.26_linux.run
安装过程中,先会让你阅读协议,按q,输入accept,接下来会让你选择性的安装,除了OpenGL选no 其余全部选yes,涉及路径的默认回车,如我上面截图显示。
$ sudo a+x cuda_8.0.61.2_linux-run
$ sudo sh cuda_8.0.61.2_linux-run
$ vi ~/.bashrc // 修改环境变量,加入
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
$source ~/.bashrc
$ nvcc -V
$ cd ~/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery
$ make clean && make
$ ./deviceQuery
若OK则如下显示,会有pass,否则为Fail
$ sudo service lightdm start // 重启桌面服务
查看tensorflow版本
在python环境中输入:
import tensorflow as tf
tf.__version__ (查看版本)
tf.__path__ (查看路径)
进入python,键入以下代码
import tensorflow as tf
hello = tf.constant('Hello Tensorflow')
sess = tf.Session()
print(sess.run(hello))
代码中可以加入避免警告。https://blog.csdn.net/u014797226/article/details/80148988
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
https://blog.csdn.net/u014797226/article/details/79626693
https://blog.csdn.net/u014797226/article/details/80229887 Ubuntu安装Tensorflow(GPU版)