赞
踩
ubuntu版本是22.04,现在最新版本是24.xx,截止当前,Nvidia的驱动最高还是22.04版本,不建议更新至最新版本。本部分是从0开始安装Nvidia docker的记录,若已安装Nvdia驱动,请直接跳至3。
更新apt,安装gcc、g++等
apt-get update
apt-get install g++ gcc make
0)预备条件
lspci | grep -i nvidia
nvidia-smi # 查看是否有显卡驱动,无显卡信息输出则未安装
sudo apt-get remove nvidia-* # 卸载原有的显卡驱动
1)安装方式1:
lshw -c display
ubuntu-drivers devices
shutdown -r now # 关机后立即重启
2)安装方式2:
sudo ./NVIDIA-Linux-x86_64-435.21.run -no-opengl-files -no-x-check -no-nouveau-check
3)安装成功验证:
安装nvidia-container-toolkit,下面提取了apt安装方式,其他比如yum、zypper等安装方式参考NVIDIA CONTAINER TOOLKIT ,apt安装方式根据下面命令复制安装即可。
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
apt-get install -y nvidia-container-toolkit
docker安装方式
apt install docker.io
docker load -i nvdocker.tar # 其中nvdocker.tar是镜像名称,需要下载
配置nvidia-docker
nvidia-ctk runtime configure --runtime=docker
systemctl restart docker
验证nvidia docker是否可用
docker run hello-world
如果输出"Hello from Docker! xxx…",表明安装成功
使用GPU
docker run有个参数 --gpus,可指定分配几个GPU
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。