1、下载对应的CUDA
前提nvidia驱动已安装好,查看教程一:
查看cuda和驱动的对应(如图):官方cuda和驱动对应
我安装的驱动是440.100,所以应选择CUDA10.2.89。到CUDA下载,拉到页面最底下选择download now,可以看到旧的cuda版本(如图)。
2、安装CUDA
quick start guide
选择CUDA Toolkit10.2,看到下载页面如下,在终端输入提示的命令即可。
注意如果出现installation failed,则需要先修改run文件权限:
sudo chmod a+x cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
然后就开始进入安装,在一大堆文字后输入accept表示接受协议。
提示incomplete installation这里不用管。
配置cuda环境变量:
sudo su //切换为root超级用户的命令
sudo gedit ~/.bashrc //打开.bashrc文件
//在~/.bashrc的最后添加:
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
添加之后一定要更新:source ~/.bashrc
3、测试CUDA是否安装成功
nvcc --version
成功后显示
执行一个samples中的程序检测一下:
cd /home/hy/NVIDIA_CUDA-10.2_Samples/1_Utilities/deviceQuery //输入自己的samples文件夹路径
sudo make
./deviceQuery
最后结果result=PASS应该就算成功了。
4、安装cudnn
首先到官网下载cuDNN https://developer.nvidia.com/cudnn ,需要注册为开发者才能下载。选择对应cuda版本的cudnn。我的是cuda10.2,故选择cudnn8,点击cudnn library for linux(x86),下载cudnn-xxxx.tgz文件。
解压文件:
tar xvzf cudnn-10.2-linux-x64-v8.0.1.13.tgz
解压完成后得到一个cuda文件夹,该文件夹下有include和lib64两个文件夹。切换到cuda/include路径下,进行以下操作:
sudo cp cudnn.h /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
切换到cuda/lib64:
sudo cp lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
5、检查cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 1
–
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include “driver_types.h”