目录

参考

前置条件

1. 检查需要安装的cuda版本

2. 英伟达官网下载对应cuda版本

3. 输入对应指令

4. 配置环境

5. 检查安装成功


参考

【CUDA】Ubuntu系统如何安装CUDA保姆级教程(2022年最新)_ubuntu安装cuda-CSDN博客

前置条件

安装好显卡驱动,参见Ubuntu22.04 安装显卡驱动_ubuntu2204安装nvidia显卡驱动-CSDN博客

1. 检查需要安装的cuda版本

nvidia-smi

需要安装cuda12.2版本。

2. 英伟达官网下载对应cuda版本

CUDA Toolkit Archive

如下图选择runfile(local)

3. 输入对应指令

wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
sudo sh cuda_12.2.2_535.104.05_linux.run

因为之前安装过显卡驱动,所以不用选驱动

(ooo上面的图配错了,Kernel Objects不选,选了之后会安装failed)

之前安装过一个版本的cuda,所以选择update

安装完成:

4. 配置环境

sudo gedit ~/.bashrc

在文件末尾添加(自己对应版本的cuda)

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

更新环境

source ~/.bashrc
sudo ldconfig

5. 检查安装成功

nvcc -V

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐