ubuntu16.04安装gpu版本的tensorflow
来源:互联网 发布:单机游戏编程语言 编辑:程序博客网 时间:2024/04/29 14:49
过程参考了http://blog.csdn.net/continueoo/article/details/71404066。
安装NVIDIA驱动
禁用自带的nouveau NVIDIA驱动
使用如下命令创建文件
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
文件中添加如下内容:
blacklist nouveauoptions nouveau modeset=0
使用如下命令更新:
sudo update-initramfs -u
重启系统。使用如下命令确认mouveau已经禁用。
lsmod | grep nouveau
输出内容为空。说明成功。
禁用lightdm桌面管理器
sudo /etc/init.d/lightdm stop
使用Ctrl+ Alt + F1
进入命令行界面。
安装NVIDIA驱动
切换到下载的NVIDIA驱动的目录下。运行如下命令
sudo ./Your-Nvidia-Driver.run –no-opengl-files
后面这个参数是用来解决重复登录的问题的。
启用lightdm桌面管理器
sudo /etc/init.d/lightdm start
安装CUDA® Toolkit 8.0
下载
从网站https://developer.nvidia.com/cuda-downloads下载合适版本的cuda toolkit。
安装
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.debsudo apt updatesudo apt install cuda
修改环境变量
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
验证安装成功
cat /proc/driver/nvidia/version
出现如下内容
nvcc -V
出现如下内容
其实没有验证完。
安装cuDNN v6
下载
从网站https://developer.nvidia.com/rdp/cudnn-download下载下面要安装的包
安装runtime library
sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb
安装developer library
sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
安装code samples和cuDNN Library User Guide
sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb
验证
cp -r /usr/src/cudnn_samples_v7/ $HOMEcd $HOME/cudnn_samples_v7/mnistCUDNNmake clean && make ./mnistCUDNN
出现test passed
信息。
安装libcupti-dev library
sudo apt-get install libcupti-dev
安装gpu版本的tensorflow
sudo pip install tensorflow-gpu
结束。
阅读全文
0 0
- ubuntu16.04安装gpu版本的tensorflow
- Ubuntu16.04+Anaconda 安装GPU版本tensorflow
- tensorflow:ubuntu16.04 gpu版本安装
- TensorFlow(gpu版本)安装教程 Ubuntu16.04
- ubuntu16.04安装tensorflow-gpu
- Ubuntu16.04下安装配置了tensorflow GPU版本后导致的常见错误
- ubuntu16.04 tensorflow-gpu版本安装好后,简单的检测代码
- Ubuntu16.04 配置tensorflow gpu版本
- ubuntu16.04安装+cuda8.0+cudnn5.1+MXNET gpu版本安装+tensorflow gpu版本安装+chainerGPU版本安装
- ubuntu16.04下安装CUDA cuDNN及tensorflow-gpu版本及caffe-gpu过程(初版)
- ubuntu16.04下安装CUDA cuDNN及tensorflow-gpu版本及caffe-gpu过程
- ubuntu16.04下安装TensorFlow(GPU加速)
- Ubuntu16.04安装tensorflow-gpu版
- Ubuntu16.04 安装 TensorFlow GPU--cuda,cudnn
- ubuntu16.04 下caffe+tensorflow+GPU 安装
- Ubuntu16.04+Tensorflow(GPU)安装教程
- Ubuntu16.04 源码安装tensorflow(GPU版)
- ubuntu16.04 安装CUDA 8.0 和 cuDNN 5.1 /cudnn6.0,可适用于gpu版本的(tensorflow,caffe,mxnet)
- PHP面向对象深入理解之四(设计模式-观察者模式)
- 【置换群+贪心】51Nod1125[交换机器的最小代价]题解
- 认知,构建个人的知识体系(上)
- 1.简介
- python猜数字,只有3次机会
- ubuntu16.04安装gpu版本的tensorflow
- ngui UIInput输入汉字的那些坑
- 小端格式和大端格式(Little-Endian&Big-Endian)
- 七夕
- UVA
- python爬虫 (一)
- highcharts学习历程,包括对同类型的echarts,google charts对比比较记录
- TensorFlow学习(1)
- speex NS(noise suppression)函数调用 自我学习