2024-11-09 07:25:03 +08:00
# gpu
2024-11-09 06:33:02 +08:00
English | [中文 ](README_CN.md )
**THIS doc for non-CN USER**
**dockerhub: https://hub.docker.com/r/ben0i0d/gpu**
**Our Gitea: https://eoelab.org:1027/ben0i0d/gpu**
2024-11-09 07:25:03 +08:00
## 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
2024-11-09 06:33:02 +08:00
## Upstream
* NVIDIA: https://gitlab.com/nvidia/container-images/cuda
* ROCM: https://github.com/ROCm/ROCm-docker
## Platform
2024-11-09 07:25:03 +08:00
* OS : debian 12 / ubuntu 24.04
2024-11-09 06:33:02 +08:00
* ARCH : x86_64
## Version
* cuda: 12.4
* rocm: 6.2
2024-11-09 07:25:03 +08:00
## Usage
* ROCM `docker run --device /dev/kfd --device /dev/dri --security-opt seccomp=unconfined IMAGE`
## Mirror source
* debian ustc：
* ubuntu ustc：
