Jetson docker环境搭建
1. 安装必要依赖
sudo apt update
sudo apt install -y curl gnupg
sudo docker pull nvcr.io/nvidia/l4t-base:r35.4.1
NVIDIA镜像查找链接:
https://catalog.ngc.nvidia.com/orgs/nvidia/containers
2. 添加 NVIDIA GPG 密钥
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
3. 添加仓库(注意:使用 stable,不是 experimental)
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \sed 's#deb https://#deb [arch=arm64 signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
4. 安装 toolkit
sudo apt update
sudo apt install -y nvidia-container-toolkit
2. 配置 Docker 使用 nvidia runtime
自动生成 /etc/docker/daemon.json 配置
sudo nvidia-ctk runtime configure --runtime=docker
你应该看到类似输出:
I! Configuring container runtime...I! Writing configuration to /etc/docker/daemon.json
3. 重启 Docker 服务
sudo systemctl restart docker
4. 验证 nvidia runtime 是否可用
docker info | grep -A5 "Runtimes"输出带nvidia就可以了。
5. 现在可以运行容器了(极简命令)
sudo docker run -it --rm --runtime=nvidia --network host nvcr.io/nvidia/l4t-jetpack:r35.4.1 bash