debian安装NVIDIA显卡驱动并启动DOCKER支持
#!/bin/bash set -e tmp_dir="/tmp" nouveau_loaded=$(lsmod | grep nouveau | wc -l) if [ "$nouveau_loaded" -eq 0 ]; then echo "Nouveau driver is already disabled." else echo "blacklist nouveau" | tee -a /etc/modprobe.d/blacklist.conf echo "options nouveau modeset=0" | tee -a /etc/modprobe.d/blacklist.conf reboot_server=true fi apt-get update && apt-get install -y curl linux-headers-$(uname -r) cd ${tmp_dir} curl -O https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.1-1_all.deb dpkg -i cuda-keyring_1.1-1_all.deb apt-get update && apt-get -y install cuda-drivers cuda-toolkit-12-3 rm -rf ${tmp_dir}/cuda-keyring_1.1-1_all.deb curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | 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' | \ tee /etc/apt/sources.list.d/nvidia-container-toolkit.list sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list apt-get update && apt-get install -y nvidia-container-toolkit nvidia-ctk runtime configure --runtime=docker if [ "$reboot_server" = true ]; then echo "" echo "**********************************************************" echo "* *" echo "* IMPORTANT: Please reboot your server to apply changes! *" echo "* *" echo "**********************************************************" echo "" fi 验证是否安装成功
宿主机验证：
nvidis-smi 
docker容器验证
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi 
原文链接：https://blog.csdn.net/weixin_53740389/article/details/136420191?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522779804426f5e18bb8a4d6b0401249e45%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=779804426f5e18bb8a4d6b0401249e45&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_ecpm_v1~times_rank-9-136420191-null-null.nonecase&utm_term=%E9%A3%9E%E7%89%9BOS
© 版权声明
声明📢本站内容均来自互联网，归原创作者所有，如有侵权必删除。
本站文章皆由CC-4.0协议发布，如无来源则为原创，转载请注明出处。
THE END
