本文主要介绍在Ubuntu系统中,如何替换官方源为中科大(USTC)镜像源,安装nvidia-container-toolkit工具,解决官方源下载速度慢、下载失败的问题。该工具主要用于让Docker等容器能够正常调用NVIDIA GPU,适用于需要在容器中运行GPU加速任务的场景。
以下步骤均需在终端中执行,可直接复制命令运行,确保终端拥有sudo权限(管理员权限)。
更新软件源并安装curl、gnupg2依赖工具,用于后续导入密钥和获取镜像源列表:
bashsudo apt-get update && sudo apt-get install -y --no-install-recommends curl gnupg2
导入中科大镜像源的GPG密钥,用于验证软件包的合法性,避免安装恶意软件:
bashcurl -fsSL https://mirrors.ustc.edu.cn/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
获取中科大镜像源的列表文件,并替换系统中原有的nvidia-container-toolkit源列表,确保后续安装从国内镜像源下载:
bashcurl -s -L https://mirrors.ustc.edu.cn/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://nvidia.github.io#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://mirrors.ustc.edu.cn#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
替换镜像源后,再次更新软件源,使系统识别新的镜像源:
bashsudo apt-get update
固定安装1.18.2-1版本,避免因版本过高或过低导致的兼容性问题,命令如下:
bashexport NVIDIA_CONTAINER_TOOLKIT_VERSION=1.18.2-1
sudo apt-get install -y nvidia-container-toolkit=${NVIDIA_CONTAINER_TOOLKIT_VERSION} nvidia-container-toolkit-base=${NVIDIA_CONTAINER_TOOLKIT_VERSION} libnvidia-container-tools=${NVIDIA_CONTAINER_TOOLKIT_VERSION} libnvidia-container1=${NVIDIA_CONTAINER_TOOLKIT_VERSION}
nvidia-container-toolkit --version命令,验证安装是否成功(显示1.18.2版本即为安装成功)。本文作者:苏皓明
本文链接:
版权声明:本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!