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