本文环境 ubuntu 18.04

1 nvidia显卡驱动

sudo apt-get install nvidia-driver-435

2 cuda

2.1 下载cuda

CUDA官网

选择配置，但是Installer Type必须选择runfile,根据下方给出的url下载安装包，可选择使用wget直接在ubuntu下载，也可以选择使用其他方式下载然后上传至ubuntu

2.2 安装cuda

在ubuntu上运行下方代码

sh cuda_10.0.130_410.48_linux.run

根据下方步骤进行安装

1. 文档，输入q
2. 接受，输入accept，回车
3. 是否安装驱动，输入n，回车
4. 是否安装cuda，输入y，回车
5. 默认安装目录，回车
6. 是否建立目录软链接，回车
7. 是否安装默认程序，输入y
8. 默认程序位置，回车

2.3 配置cuda

vim /etc/profile
加入下面三行环境变量
export CUDA_HOME=/usr/local/cuda:/usr/local/cuda-10.0
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/root/TensorRT-6.0.1.5/lib:/usr/local/cuda/extras/CUPTI/lib64
export PATH=$PATH:$CUDA_HOME/bin 
保存
source /etc/profile

3 cudnn

3.1 下载cudnn

cudnn官网,需要登录下载

选择跟cuda相对应的版本，选择cuDNN Library for Linux版本下载

3.2 安装cudnn

3.2.1 解压

tar -zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

3.2.2 安装

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*

3.3 查看版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

• ← 上一篇
• 下一篇 →