Ubuntu16.04 下cuda的安装

cuda:CUDA(Compute Unified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构（ISA）以及GPU内部的并行计算引擎。
cuDNN:CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案，目前支持的DL库包括Caffe，ConvNet,Torch7等。

cuda 的安装

cuda下载官网：https://developer.nvidia.com/cuda-downloads
按照如下的配置下载

下载完成后再执行下述操作

sudo dpkg -i@H_301_21@ cuda-repo@H_301_21@-ubuntu1604@H_301_21@-@H_301_21@8@H_301_21@-@H_301_21@0@H_301_21@-local@H_301_21@-ga2_8@H_301_21@.0@H_301_21@.61@H_301_21@-@H_301_21@1@H_301_21@_amd64.@H_301_21@deb
sudo apt-get@H_301_21@ update
sudo apt-get@H_301_21@ install cuda

检测cuda是否安装成功

配置环境变量

sudo@H_301_21@ vim ~/.bash_profile

# 加入@H_301_21@

export@H_301_21@ LD_LIBRARY_PATH="$LD_LIBRARY_PATH@H_301_21@:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"@H_301_21@
export@H_301_21@ CUDA_HOME=/usr/local/cuda

# 保存后执行@H_301_21@

source@H_301_21@ ./bash_profile

执行nvidia-smi，如果出现下述界面则安装成功
遇到的问题
- Failed to initialize NVML: Driver/library version mismatch：重启就可以解决。

CuDNN 的安装

进入网址，注册登录后方能下载 https://developer.nvidia.com/cudnn
按照下图所示，进行操作

下载完成后执行下述操作：

tar xvzf cudnn-8.0@H_301_21@-linux-x64-v5.1@H_301_21@-ga.tgz
sudo cp cuda/include@H_301_21@/cudnn.h /usr/local@H_301_21@/cuda/include@H_301_21@
sudo cp cuda/lib64/libcudnn* /usr/local@H_301_21@/cuda/lib64
sudo chmod a@H_301_21@+r /usr/local@H_301_21@/cuda/include@H_301_21@/cudnn.h /usr/local@H_301_21@/cuda/lib64/libcudnn*

查看cuda的版本
```
nvcc -V@H_301_21@
```
注意事项
- cudnn版本问题：tensorflow-gpu 1.2 支持的cudnn版本是5.1 如果两者不匹配，则无法运行。所以运行前必须将两者版本匹配

Ubuntu16.04 下cuda的安装

猜你在找的Ubuntu相关文章