170816 Ubuntu-Nvidia CUDA+Driver+cuDNN installation-Titan Xp
原创
©著作权归作者所有:来自51CTO博客作者GuokLiu的原创作品,请联系作者获取转载授权,否则将追究法律责任
参考文献:
Ubuntu 16.04 + Nvidia 显卡驱动 + Cuda 8.0 (问题总结 + 解决方案)
深度学习Keras快速开发入门
前言血泪史: 除非有完全把握,不建议安装最新驱动,可通过下面代码查看适合自己电脑的Nvidia驱动。网上大部分说CUDA选取.run文件安装,然而我却碰到各种问题最终用.deb文件安装,终结就是适合自己的才是最好的!希望对后面碰到问题的童鞋有所帮助!
sudo apt-cache search nvidia*
Step1: Ubuntu 环境的设置
安装开发包,打开终端输入:
# 系统升级
sudo apt update
sudo apt upgrade
# 安装Pyhton基础开发包
sudo apt install -y libopenblas-dev liblapack-dev libatlas-base-dev
安装运行加速库,打开终端输入:
sudo apt install -y libopenblas-dev liblapack-dev libatlas-base-dev
Step2: 安装CUDA开发环境
CUDA 8.0 下载地址:https://developer.nvidia.com/cuda-downloads
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
安装好后讲CUDA路径添加至环境变量
sudo gedit /etc/bash.bashrc
在bash.bashrc中添加一下内容
export CUDA_HOME=/usr/local/cuda-8.0
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 gedit /etc/bash.bashrc
终端输入:
在bashrc中输入以上相同内容
测试是否安装成功?在终端输入:
如果成功则会出现相应的信息,此时需要重新启动!
官方测试方法:
# Install a writable copy of the samples then build and run the nbody sample:
$ cuda-install-samples-8.0.sh ~ #安装测试样本
$ cd ~/NVIDIA_CUDA-8.0_Samples/5_Simulations/nbody #切换到测试样本目录
$ make #制造测试文件
$ ./nbody #执行测试
Step3:安装加速库cuDNN文件
https://developer.nvidia.com/cudnn
四步走:注册-下载-解压-复制
注册:下载cuDNN需要Nvidian账号
下载:下载自己需要的版本(我的是的dev版本,run版本貌似安装有问题,不能verify测试)
解压:
tar -zxvf cudnn-8.0-linux-x64-v7.solitairetheme8
复制:
sudo cp /cuda/include/cudnn.h /usr/local/cuda/include
sudo cp /cuda/lib64/* /usr/local/cuda/lib64/
Step4:最终安装成功!
三个检测:
nvcc -V # CUDA 安装成功
nvidia-smi # 显卡安装成功
./ nbody # 测试文件可以运行