- 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 cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb sudo apt-get update sudo apt-get install cuda
- 检测cuda是否安装成功
- 配置环境变量
sudo vim ~/.bash_profile
# 加入
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
# 保存后执行
source ./bash_profile
- 执行nvidia-smi,如果出现下述界面则安装成功
- 遇到的问题
- Failed to initialize NVML: Driver/library version mismatch:重启就可以解决。
- 配置环境变量
- CuDNN 的安装
- 进入网址,注册登录后方能下载 https://developer.nvidia.com/cudnn
- 按照下图所示,进行操作
- 下载完成后执行下述操作:
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*
ubuntu 16.04 下cuda的安装
最新推荐文章于 2020-11-19 11:03:28 发布