ToB企服应用市场:ToB评测及商务社交产业平台

标题: Ubuntu 20.04下多版本CUDA的安装与切换 超详细教程 [打印本页]
作者: 写过一篇    时间: 2024-7-29 23:59
标题: Ubuntu 20.04下多版本CUDA的安装与切换 超详细教程
前言

当我们复现代码时，总会需要不同版本的 CUDA 。此时，我们就需要安装多个版本的 CUDA ，并实现机动切换。
本文将在已有 CUDA11.3 的 Ubuntu20.04 下，安装 CUDA11.8 。

一、安装 CUDA

1.找到所需版本对应下令

CUDA官方下载地址
1.1 找到对应版本

1.2 选择对应的操纵系统
Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile（local）

可以看到有两条下令，依次对应下载和安装。

2.下载 .run 文件

  1. 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 文件进行安装
  1. su
  2. do 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) 打开设置文件
  1. su
  2. do gedit ~/.bashrc
复制代码

(2) 写入所需的环境变量
  1. # CUDA enviroment variable
  2. export PATH=$PATH:/usr/local/cuda/bin
  3. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
  4. export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
复制代码

(3) 更新环境变量
  1. source ~/.bashrc
复制代码
4.2 软连接

(1) 进入 /usr/local 目录下
  1. cd /usr/local
复制代码


(2) 删除原有的软连接
  1. su
  2. do rm -rf cuda
复制代码

(3) 新建软连接，将其连接到 CUDA11.8
  1. su
  2. do ln -s /usr/local/cuda-11.8 /usr/local/cuda
复制代码

5.验证安装

  1. nvcc -V
复制代码
可以看到 CUDA 版本已经更新为 11.8

二、安装 cudnn

1.下载 cudnn

CUDA官方网址
选择得当自己 CUDA 版本的 cuDNN 下载


2.解压文件

(1) 进入下载目录
(2) 解压 .xz
  1. xz -d cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
复制代码

(3) 解压 .tar
  1. tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar
复制代码

3.更换文件

(1) 切换超级用户
  1. su
复制代码
(2) 复制 lib 或 lib64 文件夹下的文件到 CUDA 安装位置的 lib64 文件夹下
  1. cp lib/* /usr/local/cuda-11.8/lib64/
复制代码

(3) 复制 include 文件夹下的全部文件到 CUDA 安装位置的 include 文件夹下
  1. cp include/* /usr/local/cuda-11.8/include/
复制代码

4.验证安装

  1. cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
复制代码
出现下图提示即为安装成功

三、切换 CUDA 版本

1.切换版本

(1) 进入 /usr/local 目录下
  1. cd /usr/local
复制代码

(2) 删除软连接
  1. su
  2. do rm -rf cuda
复制代码

(3) 新建软连接到想切换的版本
  1. su
  2. do ln -s /usr/local/cuda-11.3 /usr/local/cuda
复制代码

2.查抄版本

  1. nvcc -V
复制代码
可以看到版本已经切换成功


免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4