gpu

English | 中文

THIS doc for non-CN USER

dockerhub: https://hub.docker.com/r/ben0i0d/gpu

Our Gitea: https://eoelab.org:1027/ben0i0d/gpu

Prerequisites

• ROCM : amdgpu-dkms -> Docker containers share the kernel with the host OS. Therefore, the ROCm kernel-mode driver (amdgpu-dkms) must be installed on the host.
• CUDA : nvidia-container-toolkit/nvidia-docker2

Upstream

• NVIDIA: https://gitlab.com/nvidia/container-images/cuda
• ROCM: https://github.com/ROCm/ROCm-docker

Platform

• OS : debian 12 / ubuntu 24.04
• ARCH : x86_64

Version

• cuda: 12.4
• rocm: 6.2

Usage

• ROCM docker run --device /dev/kfd --device /dev/dri --security-opt seccomp=unconfined IMAGE

Mirror source

• debian ustc：https://mirrors.ustc.edu.cn/help/debian.html
• ubuntu ustc：https://mirrors.ustc.edu.cn/help/ubuntu.html