Ubuntu安装CUDA

最新推荐文章于 2025-04-14 21:07:59 发布

èé

最新推荐文章于 2025-04-14 21:07:59 发布

阅读量159

点赞数 3

文章标签： ubuntu linux 运维

本文链接：https://blog.youkuaiyun.com/weixin_43392432/article/details/146254399

版权

1、检查Ubuntu版本：

lsb_release -a # 确保输出为Ubuntu 20.04 LTS（Focal Fossa）

确认驱动是否安装：

dpkg -l | grep -i nvidia  #如果无输出则驱动未安装;有输出且状态为ii(已安装)

2、卸载现有Nvidia驱动和CUDA：

sudo apt purge nvidia-* cuda-* libnvidia-*
sudo apt autoremove
sudo rm /etc/apt/sources.list.d/cuda*.list
sudo apt update

3、禁用Nouveau驱动（如果未禁用的话）：

sudo nano /etc/modprobe.d/blacklist-nouveau.conf

# 可以找到modprobe.d文件的位置，一般都在/etc/下面，打开它
sudo nano modprobe.d
# 在modprobe.d中添加下面两行代码：
blacklist nouveau
options nouveau modeset=0

# 添加完按CTRL+O保存，然后按enter确认，再按CTRL+X退出编辑器

4、更新并重启： sudo update-initramfs -u sudo reboot

sudo update-initramfs -u 
sudo reboot

5、安装CUDA Toolkit和驱动：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4
sudo apt-get install -y nvidia-open

6、再次确认驱动是否安装：

dpkg -l | grep -i nvidia # 如果无输出则驱动未安装;有输出且状态为ii(已安装)

7、配置环境变量：

# 编辑~/.bashrc：
nano ~/.bashrc      
# 在文件末尾添加： 
export PATH=/usr/local/cuda-12.4/bin:$PATH 
exprt LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH     
(保存退出，CTRL+O,enter,CTRL+X)

8、重启并检查驱动和CUDA：