步骤1:CUDA官网选择适合自己系统的版本下载。
我的系统是Ubuntu18.04、64位,选择CUDA10版本如下
步骤2:安装CUDA 在下载文件的 目录里右键在终端打开
在命令行中键入:
sudo sh cuda_10.0.130_410.48_linux.run
一直按Enter直至把声明读完(或者按ctrl c)
如果驱动是独立安装了,一定要选择不安装驱动!选择如下:
步骤3:
添加环境变量:终端输入 sudo gedit ~/.bashrc 打开环境变量文件 将变量添加到文件底部
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并退出 输入命令生效环境:
source ~/.bashrc
测试是否安装成功
上面的选项选择安装了CUDA例子,运行其中一个来测试是否安装成功:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
结果如下就表明成功啦
步骤4:安装CUDNN
在官网下载安装包,需要注册登录才能下载。选择适合自己的版本
下载完成后解压并进入文件夹:命令输入
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*
在终端查看CUDNN版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
步骤5:添加cudnn软连接 终端输入6行命令
sudo cp /usr/local/cuda-10.0/lib64/libcudart.so.10.0 /usr/local/lib/libcudart.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcublas.so.10.0 /usr/local/lib/libcublas.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcurand.so.10.0 /usr/local/lib/libcurand.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcusolver.so.10.0 /usr/local/lib/libcusolver.so.10.0 && sudo ldconfig
sudo cp /usr/local/cuda-10.0/lib64/libcufft.so.10.0 /usr/local/lib/libcufft.so.10.0 && sudo ldconfig
结束