34 lines
851 B
Markdown
34 lines
851 B
Markdown
# 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
|
||
|
||
|