【CUDA】Ubuntu系统如何安装CUDA保姆级教程(2022年最新)

自牧君 2024-06-22 12:01:02 阅读 67

本期目录

Linux安装CUDA

Linux安装CUDA

输入以下命令,查看 GPU 支持的最高 CUDA 版本。笔者这里显示的是 11.6 ,这意味着,安装的 CUDA 版本必须 <= 11.6 。

nvidia-smi

image-20221016112555192

前往 Nvidia 的 CUDA 官网:CUDA Toolkit Archive | NVIDIA Developer ,笔者下载 CUDA 11.6.2 版本:

image-20221016113052516

如下图选择:

image-20221016113304942

选择好后,下方会显示两条 Linux 命令。在 Linux 终端输入这条命令:

$ wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run

这里笔者的服务器无法翻墙,直接在服务器上下载会超时。所以直接复制下载地址,先在 Windows 系统下下载好,再上传到 /root/ 目录下。

https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run

image-20221016115213397

这里由于笔者没有 root 权限,无法上传到 /root/ 目录下。因此将此安装包上传至 /home/og/xxx 自己的工作目录下。输入以下命令:

$ sudo sh cuda_11.6.2_510.47.03_linux.run

弹出来第一个选择框,笔者这台服务器已经安装过 Nvidia 的显卡驱动了,因此选择 “continute” ,会出现下面这个对话框,选择 “accept” 。

image-20221016132927190

笔者服务器已经安装过 Nvidia 显卡驱动了,因此不需要再安装了。如果你尚未安装驱动,可以顺便一起安装了。摁一下空格取消 Driver 安装,直接选择 Install 安装:

image-20221016134316003

安装成功:

image-20221016134822770

image-20221016134927175

接下来配置环境变量,直接在 Linux 命令行输入以下命令:

sudo vim ~/.bashrc

在文件末尾添加 (这里笔者假设你已经熟悉 Linux 下 Vim 文本编辑器的使用,不熟悉的强烈推荐去学习一下) :

export PATH=/usr/local/cuda-11.6/bin${ PATH:+:${ PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}export CUDA_HOME=/usr/local/cuda-11.6

更新系统环境:

source ~/.bashrcsudo ldconfig

验证 CUDA 是否安装成功,输入以下命令:

nvcc -V

输出下图即为成功安装:

image-20221016135823246



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。