CUDA安装
一、常用命令
1.查看NVIDIA 版本 nvidia-smi
2.查看CUDA版本 nvcc -V
二、下载安装相关资源
1.CUDA Toolkit CUDA Toolkit Archive | NVIDIA Developer
选择自己需要的版本
点击后进入下图界面
(1)在浏览器复制wget地址,进行下载run文件。
(2)下载完成后,在下载文件所在目录打开终端,输入 sudo sh cudaxxx.run(下载文件名)
(3)提示是否接受
输入accept
(4)取消勾选Driver(回车取消)
(5)光标下移至Install,回车,等待安装即可
(6)配置相关环境
(6)1.修改文件
sudo gedit ~/.bashrc
打开/.bashrc文件,输入下列内容至文件尾部
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
(6)2. 更新配置
source ~/.bashrc
(6)3.测试安装结果
cd /usr/local/cuda-11.3/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
结果出现 Result = Pass,即为安装成功
2.CUDNN cuDNN Archive | NVIDIA Developer
(1)打开下载文件所在目录
(2)解压:
tar -zxvf 文件名.tgz
(3)运行下列命令 ,配置文件
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
(4)查看CUDNN版本
cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2