cudaToolkit安装
1、下载cuda-toolkit,点击下面链接
https://developer.nvidia.com/cuda-toolkit-archivehttps://developer.nvidia.com/cuda-toolkit-archive
页面跳转显示如下图:
选择需要的版本,点击,页面跳转如下图,按照图像上的标注进行选择和安装:
参考上图,安装命令如下:
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run
出现安装条款如下:
输入“accept”
选择“install”,即进行安装。
2、进行环境配置
sudo vim ~/.bashrc
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
刷新环境:
source ~/.bashrc
cuDNN安装
1. 下载cuDNN
cuDNN下载网址:
https://developer.nvidia.com/rdp/cudnn-archive
打开网址,选择适合自己系统、cuda版本的cudnn
一般选择 “tar包”,下载后,文件名如下:
cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
解压文件:
tar -xf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
然后,使用下面两条指令复制cuda文件夹下的文件到 /usr/local/cuda/lib64/ 和 /usr/local/cuda/include/ 中。
cp cuda/lib64/* /usr/local/cuda-11.0/lib64/
cp cuda/include/* /usr/local/cuda-11.0/include/
拷贝完成后,我们可以使用如下的命令查看cuDNN的信息:
cat /usr/local/cuda-11.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
至此,cudatoolkit 和 cudnn 都已安装完成,可以尽情利用GPU加速搞AI算法了