当前位置:   article > 正文

Ubuntu 22.04 安装cuda,适用20.04_ubuntu22.04可以安装cuda11.2吗

ubuntu22.04可以安装cuda11.2吗

1. 安装CUDA 驱动

驱动安装可以使用系统自带的软件,如下图一所示,‘Software & Updates’.

图一

打开以后如下图二。选择图中 ‘Additional Drivers’。会显示目前装的驱动,其中,当前显示的是NVIDIA 的驱动。下边给出的几个推荐的驱动版本,根据显卡不同会有不同。我的显卡是RTX A5000,我选择470的版本驱动。选择以后点击右下角‘Apply Changes’。可能会花费一点时间,等进度条完成后,‘Revert’左侧会出现一个新的按钮‘Restart’。此时选择‘Restart’重启即可。

图二
图二

2. 安装cuda-toolkit

进入Nvidia官网 CUDA Toolkit Archive | NVIDIA Developer  .选择合适的版本,点击,会进行跳转,我选择的是cuda 11.1. 打开后如图所示。注意此时没有ubuntu 22.04版本,但是没有关系,我们可以安装20.04版本。

正式安装之前,我们先检查电脑中是不是有gcc。 终端中输入gcc --version .  如果提示gcc 信息,则已经安装,如果提示找不到gcc命令,则使用下面命令进行安装。安装完成后检查一下是否安装成功。

  1. sudo apt-get install manpages-dev
  2. sudo apt-get update
  3. sudo apt install build-essential
  4. sudo apt-get install manpages-dev

 下面进行正式安装,如下图四是nvidia官网给出的教程。但是不需要全都输入。只需要从第三行开始输入即可。也就是下面的代码。

图四

 有一个地方需要注意,最后一行,我们将cuda 改成了cuda-11-1.因为如果只写cuda安装的是最新版本。

  1. wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers
  2. /cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
  3. sudo dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
  4. sudo apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
  5. sudo apt-get update
  6. sudo apt-get -y install cuda-11-1

 3. 添加环境路径

首先确认一下cuda在本电脑上的具体文件夹名称。输入如下代码。发现确实有cuda-11.1的文具文件夹,那么我们可以直接使用这个路径。

  1. jianchao@jianchao-ThinkPad-P15-Gen-2i:/$ cd /usr/local/
  2. jianchao@jianchao-ThinkPad-P15-Gen-2i:/usr/local$ ls
  3. bin cuda cuda-11.1 etc games include lib man sbin share src

执行如下代码,将cuda路径添加进环境。注意将cuda-11.1的文件夹名称改成自己的。

  1. export PATH="/usr/local/cuda-11.1/bin:$PATH"
  2. export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"

4. 测试

终端输入nvidia-smi, 和nvcc -V测试是否显示正确信息。注意nvidia-smi显示的cuda版本是驱动支持的最大cuda版本,并不是当前安装的cuda版本。 而nvcc -V显示的是当前安装的cuda版本。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/649450
推荐阅读
相关标签
  

闽ICP备14008679号