我要评论
目录
前言
当我们复现代码时，总会需要不同版本的 cuda 。此时，我们就需要安装多个版本的 cuda ，并实现灵活切换。
本文将在已有 cuda11.3 的 ubuntu20.04 下，安装 cuda11.8 。
一、安装 cuda
1.找到所需版本对应命令
1.1 找到对应版本
1.2 选择对应的操作系统
linux -> x86_64 -> ubuntu -> 20.04 -> runfile（local）
可以看到有两条命令，依次对应下载和安装。
2.下载 .run 文件
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
3.安装 cuda
(1) 运行 .run 文件进行安装
sudo sh cuda_11.8.0_520.61.05_linux.run
(2) 方向下键↓，选中 “continue” ，回车 “enter” 确认
(3) 输入 accept，然后回车
(4) 使用方向下键↓，移动到 driver 上，回车勾选掉 driver
由于安装的多版本 cuda ，安装第一个 cuda 的时候已经安装过驱动，所以不需要再重复安装。
(5) 使用方向下键↓，移动到 install 上，回车开始安装
(6) 之后一路默认，一直敲回车即可
出现如下图的提示即是安装成功
4.配置环境变量
4.1 写入环境变量
(1) 打开配置文件
sudo gedit ~/.bashrc
(2) 写入所需的环境变量
# cuda enviroment variable
export path=$path:/usr/local/cuda/bin
export ld_library_path=$ld_library_path:/usr/local/cuda/lib64
export cuda_home=$cuda_home:/usr/local/cuda
(3) 更新环境变量
source ~/.bashrc
4.2 软连接
(1) 进入 /usr/local 目录下
cd /usr/local
(2) 删除原有的软连接
sudo rm -rf cuda
(3) 新建软连接，将其连接到 cuda11.8
sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda
5.验证安装
nvcc -v
可以看到 cuda 版本已经更新为 11.8
二、安装 cudnn
1.下载 cudnn
cuda官方网址
选择适合自己 cuda 版本的 cudnn 下载
2.解压文件
(1) 进入下载目录
(2) 解压 .xz
xz -d cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
(3) 解压 .tar
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar
3.替换文件
(1) 切换超级用户
su
(2) 复制 lib 或 lib64 文件夹下的文件到 cuda 安装位置的 lib64 文件夹下
cp lib/* /usr/local/cuda-11.8/lib64/
(3) 复制 include 文件夹下的全部文件到 cuda 安装位置的 include 文件夹下
cp include/* /usr/local/cuda-11.8/include/
4.验证安装
cat /usr/local/cuda-11.8/include/cudnn_version.h | grep cudnn_major -a 2
出现下图提示即为安装成功
三、切换 cuda 版本
1.切换版本
(1) 进入 /usr/local 目录下
cd /usr/local
(2) 删除软连接
sudo rm -rf cuda
(3) 新建软连接到想切换的版本
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda
2.检查版本
nvcc -v
可以看到版本已经切换成功
发表评论