Ubuntu 14.04安装CUDA 8.0问题总结

我这边的电脑环境是Ubuntu 14.04.4,显卡是GTX960M

查看TensorFlow所需要的CUDA和cuDNN对应版本。

https://tensorflow.google.cn/install/source

  • CUDA的run文件中会带有显卡驱动,推荐使用CUDA自带的驱动,和它的兼容性是最好的,而我们电脑本身也带有,所以装的时候,首先进入到终端模式(Ctrl + Alt + F1)卸载掉电脑本身的,不删除的话,装上CUDA会黑屏

灵感来源于这里 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

  • 双显卡OpenGL不要装,其他全部选yes或者默认 https://www.zhihu.com/question/48027732

Ubuntu 14.04安装CUDA 8.0问题总结_第1张图片

 

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 

Ubuntu 14.04安装CUDA 8.0问题总结_第2张图片

$ cd ~/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery

$ make clean && make 

$ ./deviceQuery 

若OK则如下显示,会有pass,否则为Fail

Ubuntu 14.04安装CUDA 8.0问题总结_第3张图片

$ 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版)

 

你可能感兴趣的:(TensorFlow)