linux 1080ti显卡驱动,ubuntu安装1080ti nvidia驱动+cuda8.0+TensorFlow

主要参考官方文档

系统Ubuntu 14.04

显卡为1080ti

步骤:

1.首先安装CUDA8.0

首先准备安装包,由于deb会存在各种问题,建议采用run模式安装

这里需要两个文件

CUDA8.0: cuda_8.0.61_375.26_linux.run(1.5G,chrome下载可能会出现没下完就中断的情况= =)下载地址:https://developer.nvidia.com/cuda-downloads

显卡驱动:NVIDIA-Linux-x86_64-381.22.run

1.1安装依赖

安装编译工具:$sudoapt-get install build-essential # basic requirement

$sudo apt-get install cmake git

$sudo apt-get update    #updatesource

安装依赖项: $sudo apt-getinstall libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-devprotobuf-compiler

$sudo apt-get install --no-install-recommendslibboost-all-dev

$sudo apt-get install libopenblas-devliblapack-dev libatlas-base-dev

$sudo apt-get install libgflags-devlibgoogle-glog-dev liblmdb-dev

$sudo apt-get install python-numpy python-scipypython-matplotlib

1.2禁用nouveau驱动(未执行)

ALT+CTRL+F1,进命令行;

$sudo service lightdm stop

$sudo apt-get --purge remove nvidia-*

新建黑名单,禁止系统自带驱动:$sudo vi/etc/modprobe.d/blacklist-nouveau.conf

输入: blacklist nouveau

options nouveau modset=0

保存推出(:wq)

然后执行:$sudoupdate-initramfs –u   #更新内核

执行 $lspci | grep nouveau,查看是否有内容,没有说明禁用成功,如果有内容,就$sudo reboot

重启:$sudo reboot

重启后,在登录界面,不要登录进桌面,直接ALT+CTRL+F1进命令行

1.3安装cuda8.0

进入cuda_8.0.61_375.26_linux.run所在目录

执行:

$sudo chmod +x cuda_8.0.61_375.26_linux.run

$sudo ./ cuda_8.0.61_375.26_linux.run

按q键退出RELU文档,按照如下选择,显卡驱动一定要选n,不装

其他都默认yes

最后,配置环境变量,直接放在系统配置文件profile里面:

$sudo gedit /etc/profile

在最后面加入两行代码:

export PATH=/usr/local/cuda-8.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

保存退出.

执行:$sudo ldconfig

1.4 显卡驱动安装

进入显卡驱动目录

$sudo ./ NVIDIA-Linux-x86_64-381.22.run

一路按照提示选择安装(全yes),主要有接受协议,在系统内核注册,用新路径注册,更新X-server,安装完成后会自动回到命令行

最后会出现一个kernel的错误,但是应该不影响

需要重启启动驱动

重启电脑:$sudo reboot

1.6检查是否安装成功

$nvcc –version

查看CUDA ,应该显示8.0,如果没有添加path会显示系统之前的版本

$nvidia-smi 查看驱动信息

在根目录下会出现NVIDIA_CUDA-8.0_Samples,打开终端,进入该目录:

$sudo make –j8  #编译samples,我电脑8线程,全开编译

等待2分钟左右,编译完成,执行下条指令:

$sudo ./1_Utilities/deviceQuery/deviceQuery

出现提示显卡驱动版本等信息的内容,表示安装成功

1.6 Atlas安装

sudo apt-get install libatlas-base-dev

1.7cuDNN安装

准备安装文件,cudnn的安装就是将include和lib文件拷贝到系统的cuda路径里

cudnn-8.0-linux-x64-v5.1.tgz

进入安装文件所在目录

$tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz

$cd cuda

$sudo cp lib64/lib* /usr/local/cuda/lib64/

$sudo cp include/cudnn.h/usr/local/cuda/include/

更新软连接,这里最好ls一下lib64的内容,防止版本不一致:

$cd /usr/local/cuda/lib64/

$sudo chmod +r libcudnn.so.5.1.10

$sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5

$sudo ln -sf libcudnn.so.5 libcudnn.so

更新设置:

$sudo ldconfig

Tensorflow需要CUDA_HOME 变量

在/etc/profile最后加上export CUDA_HOME=/usr/local/cuda-8.0

然后 $source /etc/profile

1.8 安装tensorflow

$sudo apt-getinstall libcupti-dev

采用系统自带的pip安装

$sudo pipinstall tensorflow-gpu

安装完成后

Python

>>>importtensorflow

这个时候出现了error

>>>import tensorflow

Traceback (mostrecent call last):

File "", line 1, in

File"/usr/local/lib/python2.7/dist-packages/tensorflow/__init__.py", line24, in

from tensorflow.python import *

File"/usr/local/lib/python2.7/dist-packages/tensorflow/python/__init__.py",line 52, in

from tensorflow.core.framework.graph_pb2import *

File"/usr/local/lib/python2.7/dist-packages/tensorflow/core/framework/graph_pb2.py",line 9, in

from google.protobuf import symbol_databaseas _symbol_database

File"/usr/local/lib/python2.7/dist-packages/google/protobuf/symbol_database.py",line 176, in

_DEFAULT =SymbolDatabase(pool=descriptor_pool.Default())

AttributeError:'module' object has no attribute 'Default'

发现是protobuf版本的问题

执行: $sudo pip install--upgrade protobuf

更新后就可以了

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值