您的位置:首页 > 其它

Ubuntu14.04 + GTX1070 + Cuda8.0 + TensorFlow安装配置

2017-06-23 09:27 405 查看
1.英伟达官网下载对应型号的显卡驱动 http://www.nvidia.cn/Download/index.aspx?lang=cn
2.安装nvidia驱动
首先,关闭图形界面显示器
sudo stop lightdm
orsudo service lightdm stop
然后,切换到nvidia驱动所在的目录
sudo sh NVIDIA-Linux-x86_64-375.26.run
安装过程中的各种选项全都接受,Accept Continue installation  OK  等等。
打开图形界面显示器
sudo start lightdmorsudo service lightdm start
验证安装是否成功
cat /proc/driver/nvidia/version
会看到NVRM version 和 GCC version
3.下载Cuda 8.0 https://developer.nvidia.com/cuda-downloads
下载类型选额runfile(local),deb也可以
4.安装Cuda8.0
先安装依赖库
sudo apt-get install freeglut3-dev
sudo apt-get install build-essential
sudo apt-get install libx11-dev
sudo apt-get install libxmu-dev
sudo apt-get install libxi-dev
sudo apt-get install libglu1-mesa
sudo apt-get install libglu1-mesa-dev转到Cuda8.0下载目录
chmod +x cuda_8.0.44_linux-run
./cuda_8.0.44_linux-run
!!!Note:安装过程中会遇到Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26? 一定选择No
其余的全都accpet和yes
添加环境变量,打开.bashrc文件,在里面添加
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}然后
source ~/.bashrc验证是否安装成功
nvidia-smi
nvcc --version(version前有两个短线)
第一条是查看GPU驱动有没有和Cuda发生冲突,如果没有冲突,会显示GPU的配置信息;第二条查看Cuda安装版本。
5.安装cuDNN 5.1 https://developer.nvidia.com/cudnn 解压
tar -zxvf 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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
6.安装TensorFlow
由于TensorFlow一直在更新,建议直接参考官网,教程很清楚。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐