31 lines
860 B
Markdown
31 lines
860 B
Markdown
# gpu
|
||
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**
|
||
|
||
## 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 |