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

 

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

yaked19

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值