GPU环境配置

ubuntu基本环境配置

1. 更新包列表

sudo apt-get update
sudo apt upgrade

2. 安装英伟达显卡驱动

• 使用wget在命令行下载驱动包

wget https://cn.download.nvidia.com/XFree86/Linux-x86_64/550.100/NVIDIA-Linux-x86_64-550.100.run

上面使用的是4090版本的，对应的驱动版本为550.100

不同型号的显卡请到英伟达官网查找下载链接：https://www.nvidia.cn/Download/Find.aspx?lang=cn&QNF=1

更新软件列表和安装必要软件、依赖

sudo apt-get update
sudo apt-get install gcc

安装驱动

sudo chmod 777 NVIDIA-Linux-x86_64-550.100.run
sudo ./NVIDIA-Linux-x86_64-550.100.run

安装CUDA
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run

安装的显卡驱动版本是550.100，可以安装CUDA 12版本。

可根据自己显卡版本选择合适的CUDA版本，查询链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

授予执行权限
sudo chmod 777 cuda_12.4.0_550.54.14_linux.run
sudo ./cuda_12.4.0_550.54.14_linux.run

输入 accept 开始安装
注意这里要按 Enter 取消勾选第一个选项,因为之前已经安装了驱动
然后选择 Install 开始安装

环境变量配置
sudo vim ~/.bashrc
按 i 键进入编辑模式,在文件尾增加两行：
在文件尾中加入以下两行：

export PATH=“/usr/local/cuda-12.4/bin:$PATH”

export LD_LIBRARY_PATH=“/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH”

按 Esc 键退出编辑模式,输入 :wq 保存并退出

更新环境变量
source ~/.bashrc

检测CUDA是否安装成功
nvcc -V