# gpu English | [中文](README_CN.md) **为中国用户的文档** **dockerhub: https://hub.docker.com/r/ben0i0d/gpu** **Gitea: https://eoelab.org:1027/ben0i0d/gpu** ## 先决条件 * CPU : nothing to do * ROCM : amdgpu-dkms -> Docker 容器与主机操作系统共享内核。因此,主机上必须安装 ROCm 内核模式驱动程序 ( amdgpu-dkms ) * CUDA : nvidia-container-toolkit/nvidia-docker2 ## 目标平台 * OS : debian 12 / ubuntu 24.04(只有ROCM) * ARCH : x86_64 ## 版本 * cuda: 12.4 * rocm: 6.2.4 ## 使用方法 ### container * CPU just `docker run -it` * ROCM `docker run --device /dev/kfd --device /dev/dri --security-opt seccomp=unconfined IMAGE` ### devcontainer(VScode) `cp -r .devcontainer WORKSDIR` ## 镜像依赖 ```mermaid graph LR CPU-->B{Base} CUDA-->B ROCM-->B B-->P{PROGRAM} P-->PA(C) P-->PB(Cpp) P-->PC(Java) PC-->PCA(Jdk) PC-->PCB(Jre) P-->PD(Js) P-->PE(Julia) P-->PF(Python) P-->PG(R) B-->G{GameServer}-->GA{MC}-->GAA(JAVA) GAA-->GAAA(JRE_8) GAA-->GAAB(JRE_11) GAA-->GAAC(JRE_17) GAA-->GAAD(JRE_21) G-->GB{MCSM}-->GBA(Web) GB-->GBB(Daemon) G-->GC(steam)-->GCA(steam_wine) ``` ## 镜像源 * debian ustc:https://mirrors.ustc.edu.cn/help/debian.html * ubuntu ustc:https://mirrors.ustc.edu.cn/help/ubuntu.html ## 上游 * NVIDIA: https://gitlab.com/nvidia/container-images/cuda * ROCM: https://github.com/ROCm/ROCm-docker