Ubuntu 24.04 LTS 装RTX50系显卡驱动&CUDA

配置:

CPU: AMD 5800X

主板:华硕 B550-F

显卡:RTX5090

写在前面:

老黄家新的RTX5090显卡发热真高。

安装NVIDIA显卡驱动从Ubuntu Software update检测不到。建议参考官网CUDA安装,同时将NVIDIA Driver安装好。

参考:https://docs.nvidia.com/datacenter/tesla/driver-installation-guide/index.html#ubuntu-installation

使用远程安装的方式进行,需要魔法。自行解决。

1.安装 cuda-keyring:

$ wget https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/cuda-keyring_1.1-1_all.deb
# dpkg -i cuda-keyring_1.1-1_all.deb
# apt update

如果cuda-keyring安装不上的话,需要添加Ubuntu仓库

$ wget https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/cuda-archive-keyring.gpg
# mv cuda-archive-keyring.gpg /usr/share/keyrings/cuda-archive-keyring.gpg
# echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/ /" \
    | tee /etc/apt/sources.list.d/cuda-$distro-$arch.list

2.安装Nvidia driver和CUDA

我安装的开源版本驱动。

$ apt install nvidia-open

CUDA安装之后应该是12.8 

apt install cuda-drivers

3.重启Ubuntu

$ reboot

至此不出意外的话,在终端输入

$ nvidia-smi

就可以看到对应的显卡驱动挂载成功了。

4.备注:

  • 自己的PC最好有一张亮机卡,或者CPU有集成显卡,用于亮机,RTX50的显卡可能上机之后没有驱动Ubuntu直接黑屏,什么都看不到。
  • Nvidia提供的官方可执行文件,官网下载的,会编译错误,报告内核不匹配,不知原因。
  • 主板建议更新BIOS到最新版本,如果PCIE5.0不支持,建议换成4.0通道。
  • 如果是AMD的平台,UMA Frame Buffer Size建议打开。不知原因会黑屏。
  • 新显卡和ubuntu的集成可能会在后面的更新逐步自动化,从Software updater里面应该也可以更新了。目前这种方案比较靠谱。

你可能感兴趣的:(ubuntu,linux,运维,机器学习,语言模型)