Ubuntu 安装 tensorflow-gpu 1.4 包含 CUDA 8.0 和cuDNN
硬件环境:NVIDIA GTX 980 Ti
系统环境:Ubuntu 16.04 64位
一.安装 NVIDIA驱动
- 关闭 Secure Boot
具体如何禁用 BIOS 中的 Secure Boot 要根据主板的情况。
以华硕主板的禁用方法为例:
首先进入 BIOS,然后选择 Boot ,进入 Secure Boot 界面:
确定 “OS Type” 是 “Windows UEFI”
进入 “Key Management”
选择 “Clear Secure Boot keys”
在你清除 ” Secure Boot keys” 之后,你将会有 “Install default Secure Boot keys” 选项来恢复默认的 keys 。
在你清除 ” Secure Boot keys” 之后, Secure Boot 会被自动关闭,你现在可以设置 “OS Type” 为 “Other OS”。
设置完成之后的效果:
2.禁用 nouveau driver
控制台输入命令,创建一个文件通过命令
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
并添加如下内容:
blacklist nouveau
options nouveau modeset=0
再更新一下
sudo update-initramfs -u
修改之后需要重启系统。确认下nouveau是已经禁用,可以使用命令:
lsmod | grep nouveau
3.在字符界面下安装驱动
首先添加ppa库,然后通过ppa安装显卡驱动,使用以下命令添加:
sudo add-apt-repository ppa:graphics-drivers/ppa
关闭图形化环境
首先进入 Ubuntu 系统字符界面,使用组合键 ALT+CTRL+F1 进入字符界面。
为了确保驱动能够正常安装,我们需要暂时关闭x-window服务(图形环境),在文本模式下输入命令进行关闭:
sudo service lightdm stop
安装驱动
首先通过以下命令来查看 NVIDIA VGA card model
sudo lshw -numeric -C display
可以使用命令:
ubuntu-drivers devices
来查看可以使用的驱动,如图:
输入以下命令,直接安装驱动:
sudo apt-get update && sudo apt-get install nvidia-384
进入系统,我们要看刚刚的显卡驱动是否已经正确安装成功,通过下面命令查看:
nvidia-smi
如图所示显卡驱动已经正确安装,显卡的型号是 GTX 980 Ti。
二.安装 CUDA
1.确定和 TensorFlow 对应的 CUDA 版本
TensorFlow GitHub 页面查看依赖的版本:
https://github.com/tensorflow/tensorflow/releases
2.下载和安装
因为 CUDA 8 兼容的gcc 版本是 4.9 ,但是ubuntu 16.04 默认的是 gcc 5,所以需要gcc降版本。
sudo apt-get install g++-4.9
sudo update-alternatives –install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives –install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives –install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives –install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives –install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives –set cc /usr/bin/gcc
sudo update-alternatives –install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives –set c++ /usr/bin/g++
下载地址:
https://developer.nvidia.com/cuda-downloads
因为上一步已经安装了显卡驱动,所以这里要选 no 。
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n
3.设置环境变量
配置CUDA环境变量
export PATH=”$PATH:/usr/local/cuda-8.0/bin”
export LD_LIBRARY_PATH=”/usr/local/cuda-8.0/lib64”
三.安装cuDNN
1.确定版本
TensorFlow GitHub 页面查看依赖的版本:
https://github.com/tensorflow/tensorflow/releases
2.下载
下载地址:
https://developer.nvidia.com/cudnn
3.安装
tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
四.安装 tensorflow-gpu
直接安装
pip install tensorflow-gpu
指定版本安装:
例如,要安装 tensorflow-gpu 1.4
pip install tensorflow-gpu==1.4
安装完成
参考教程:
https://blog.csdn.net/roach_zfq/article/details/53883976
https://www.cnblogs.com/xuliangxing/p/7569946.html
https://www.linuxidc.com/Linux/2016-11/137561.htm
https://blog.csdn.net/qq_30520759/article/details/78947034
低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。
持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。
转载内容版权归作者及来源网站所有,本站原创内容转载请注明来源。
- 上一篇
深入理解单例模式
初遇设计模式在上个寒假,当时把每个设计模式过了一遍,对设计模式有了一个最初级的了解。这个学期借了几本设计模式的书籍看,听了老师的设计模式课,对设计模式算是有个更进一步的认识。后面可能会不定期更新一下自己对于设计模式的理解。每个设计模式看似很简单,实则想要在一个完整的系统中应用还是非常非常难的。然后我的水品也非常非常有限,代码量也不是很多,只能通过阅读书籍、思考别人的编码经验以及结合自己的编码过程中遇到的问题来总结。 怎么用->怎么用才好->怎么与其他模式结合使用,我想这是每个开发人员都需要逾越的一道鸿沟。 本文主要内容 1 单例模式简介 1.1 定义 保证一个类仅有一个实例,并提供一个访问它的全局访问点。 1.2 为什么要用单例模式呢? 在我们的系统中,有一些对象其实我们只需要一个,比如说:线程池、缓存、对话框、注册表、日志对象、充当打印机、显卡等设备驱动程序的对象。事实上,这一类对象只能有一个实例,如果制造出多个实例就可能会导致一些问题的产生,比如:程序的行为异常、资源使用过量、或者不一致性的结果。 简单来说使用单例模式可以带来下面几个好处: 对于频繁使用的对象,可以省略...
- 下一篇
compile php with openssl on mac osx error 填坑
从源码手动编译 PHP 时出现如下错误: Default 1 2 3 4 5 6 7 8 9 10 11 12 13 14 Undefined symbols for architecture x86_64: "_PKCS5_PBKDF2_HMAC", referenced from: _zif_openssl_pbkdf2 in openssl.o "_TLSv1_1_client_method", referenced from: _php_openssl_setup_crypto in xp_ssl.o "_TLSv1_1_server_method", referenced from: _php_openssl_setup_crypto in xp_ssl.o "_TLSv1_2_client_method", referenced from: _php_openssl_setup_crypto in xp_ssl.o "_TLSv1_2_server_method", referenced from: _php_openssl_setup_crypto in xp...
相关文章
文章评论
共有0条评论来说两句吧...
文章二维码
点击排行
推荐阅读
最新文章
- CentOS7,CentOS8安装Elasticsearch6.8.6
- Docker使用Oracle官方镜像安装(12C,18C,19C)
- Red5直播服务器,属于Java语言的直播服务器
- SpringBoot2全家桶,快速入门学习开发网站教程
- CentOS8安装MyCat,轻松搞定数据库的读写分离、垂直分库、水平分库
- SpringBoot2更换Tomcat为Jetty,小型站点的福音
- Jdk安装(Linux,MacOS,Windows),包含三大操作系统的最全安装
- Hadoop3单机部署,实现最简伪集群
- CentOS关闭SELinux安全模块
- CentOS7安装Docker,走上虚拟化容器引擎之路