# gpu English | [中文](README_CN.md) **为中国用户的文档** **dockerhub: https://hub.docker.com/r/ben0i0d/gpu** **Gitea: https://eoelab.org:1027/ben0i0d/gpu** ## 先决条件 * ROCM : amdgpu-dkms -> Docker 容器与主机操作系统共享内核。因此,主机上必须安装 ROCm 内核模式驱动程序 ( amdgpu-dkms ) * CUDA : nvidia-container-toolkit/nvidia-docker2 ## 上游 * NVIDIA: https://gitlab.com/nvidia/container-images/cuda * ROCM: https://github.com/ROCm/ROCm-docker ## 目标平台 * OS : debian 12 / ubuntu 24.04 * ARCH : x86_64 ## 版本 * cuda: 12.4 * rocm: 6.2 ## 使用方法 * ROCM `docker run --device /dev/kfd --device /dev/dri --security-opt seccomp=unconfined IMAGE` ## 镜像源 * debian ustc:https://mirrors.ustc.edu.cn/help/debian.html * ubuntu ustc:https://mirrors.ustc.edu.cn/help/ubuntu.html