cudaToolkit安装
1、下载cuda-toolkit,点击下面链接
https://developer.nvidia.com/cuda-toolkit-archive
https://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算法了
本文详细介绍了如何在Linux系统上下载并安装CUDAToolkit的11.7.0版本,包括从NVIDIA官方网站获取安装包,执行安装命令,接受条款,以及配置环境变量。接着,文章讲述了cuDNN的下载和安装步骤,选择适合的版本,解压后将库文件复制到CUDA对应目录下,并验证安装成功。最后,确认CUDA和cuDNN安装完毕,可用于GPU加速的AI算法开发。
969





