Ubuntu 24.04 LTS如何安装Nvidia显卡驱动、CUDA、NVIDIA Container Toolkit套件
1、安装Nvidia显卡驱动
若有Nvidia显卡,Ubuntu系统会安装开源的nouveau驱动,用指令sudo lshw -C display确认,driver区域会显示"nouveau"。
#卸载自带的驱动 sudo apt update sudo apt upgrade sudo apt purge *nvidia* |
使用ubuntu-drivers list指令列出目前Nvidia显示卡可用的驱动版本
# 让Ubuntu自动挑选推荐的驱动版本
sudo ubuntu-drivers install # 或者手动指定版本,填入要安装的Nvidia驱动版本号。 sudo ubuntu-drivers install nvidia:570 安装后nouveau应会自动加入黑名单禁止加载。接着重新启动,用 sudo lshw -C display确认是否安装成功,driver区域应会显示 "nvidia" 。 |
www.zhangfangzhou.cn
2、双GPU显卡笔记本电脑
像Intel+Nvidia这种的双GPU笔记本电脑,即使装了Nvidia驱动也可能继续用Intel的GPU渲染3D,导致3D性能低下。
此时可以使用prime-select指令,指定用Nvidia显示卡负责渲染。
sudo prime- select nvidia 重开机后再使用指令:vulkaninfo --summary查看主显示卡为何。 |
3、Ubuntu安装cuda,CUDA Toolkit Installer。
Installation Instructions: wget https: //developer .download.nvidia.com /compute/cuda/repos/ubuntu2404/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-8 用 nvcc --version 确认cuda的版本,如果显示Command nvcc not found,则编辑~/.bashrc vim ~/.bashrc export PATH= /usr/local/cuda-12 .8 /bin ${PATH:+:${PATH}} export LD_LIBRARY_PATH= /usr/local/cuda-12 .8 /lib64 :${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} #更新变量 source ~/.bashrc # nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2025 NVIDIA Corporation Built on Fri_Feb_21_20:23:50_PST_2025 Cuda compilation tools, release 12.8, V12.8.93 Build cuda_12.8.r12.8 /compiler .35583870_0 |
www.zhangfangzhou.cn
4、安装NVIDIA Container Toolkit,这是设计给Docker和Podman容器用的Nvidia工具,使容器可以使用CUDA计算。
即使宿主机没有安装CUDA,容器內照样可以使用CUDA计算,方便你在容器里面跑不同版本的CUDA,不会受到宿主机的CUDA版本影响。
必须先安装Nvidia专有驱动才可以安装NVIDIA Container Toolkit。
(1)在Ubuntu安装Docker
(2)加入NVIDIA Container Toolkit的套件库
curl -fsSL https: //nvidia .github.io /libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring .gpg \ && curl -s -L https: //nvidia .github.io /libnvidia-container/stable/deb/nvidia-container-toolkit .list | \ sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \ sudo tee /etc/apt/sources .list.d /nvidia-container-toolkit .list 安装NVIDIA Container Toolkit sudo apt update sudo apt install nvidia-container-toolkit 向Docker注册Nvidia sudo nvidia-ctk runtime configure --runtime=docker 重新启动Docker sudo systemctl restart docker 执行Ubuntu容器,测试能否出现Nvidia显卡的信息 sudo docker run -- rm --runtime=nvidia --gpus all ubuntu nvidia-smi |
www.zhangfangzhou.cn
5、安装TensorRT,TensorRT是Nvidia推出的深度学习推理平台。
必须先安装CUDA才能安装TensorRT。 https://developer.nvidia.com/nvidia-tensorrt-download
安装TensorRT的deb档,加入套件库
# 指定系统版本 os= "ubuntu2204" # 指定TensorRT版本 tag= "10.5.0.x-1+cuda12.6" sudo dpkg -i nv-tensorrt- local -repo-${os}-${tag}_1.0-1_amd64.deb sudo cp /var/nv-tensorrt-local-repo- ${os}-${tag}/*-keyring.gpg /usr/share/keyrings/ sudo apt update 安装TensorRT sudo apt install tensorrt |
推荐本站淘宝优惠价购买喜欢的宝贝:
本文链接:https://zblog.hqyman.cn/post/10703.html 非本站原创文章欢迎转载,原创文章需保留本站地址!
休息一下~~