国内使用N卡,在容器(docker)环境中需要安装nvidia-container-toolkit,才能在docker中调用GPU进行计算。详细安装步骤可以参考官方网站的install guide:
Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.17.0 documentation
操作系统信息:
# cat /etc/os-release
PRETTY_NAME="Ubuntu 24.04.1 LTS"
NAME="Ubuntu"
VERSION_ID="24.04"
VERSION="24.04.1 LTS (Noble Numbat)"
VERSION_CODENAME=noble
ID=ubuntu
ID_LIKE=debian
HOME_URL="https://www.ubuntu.com/"
SUPPORT_URL="https://help.ubuntu.com/"
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-po