ROCm/ROCm-docker

해결하는 문제

Docker 컨테이너에는 Linux 커널이 포함되어 있지 않으며 호스트의 커널을 공유합니다. ROCm 컴퓨트 스택은 호스트 머신에 특수 커널 드라이버(ROCK-Kernel-Driver)를 설치해야 합니다. ROCm-docker는 다른 모든 ROCm 소프트웨어 레이어를 Docker 이미지에 패키징하는 방법을 제공하므로, 매번 모든 구성 요소를 수동으로 조립하지 않고도 컨테이너에서 ROCm 애플리케이션을 실행할 수 있습니다.

작동 방식

이 저장소에는 템플릿 파일에서 Dockerfile을 생성하는 설정 스크립트(setup.sh)가 포함되어 있습니다. 이 스크립트는 사전 빌드된 패키지(repo.radeon.com에서)에서 ROCm을 설치할지 소스에서 빌드할지 선택할 수 있으며, Ubuntu 기본 이미지 버전, 디버그 플래그 설정 등을 선택할 수도 있습니다. 스크립트가 실행된 후 각 하위 디렉터리에는 특정 ROCm 레이어용 이미지를 빌드하는 데 사용할 수 있는 Dockerfile이 포함됩니다. 이 프로젝트는 또한 Docker Compose를 사용하여 여러 컨테이너를 조정하고 장치 플래그(GPU 액세스 등)를 기억합니다. Docker Hub에는 빠른 시작을 위한 rocm/rocm-terminal이라는 즉시 사용 가능한 이미지가 있습니다.

대상 사용자

이것은 개발, 테스트 또는 배포를 위해 Docker 컨테이너 내에서 ROCm(AMD의 GPU 컴퓨트 스택)을 실행하려는 개발자와 사용자를 위한 것입니다. Docker 및 Linux 명령에 대한 어느 정도의 친숙함을 가정합니다. 설정 가이드와 asciicast 데모는 초보자가 시작하는 데 도움이 됩니다.

주요 기능

• 템플릿에서 사용자 정의 가능한 Dockerfile을 생성하는 설정 스크립트 제공(Ubuntu 버전, 소스 빌드, 디버그 플래그 등의 옵션 포함).
• 두 가지 설치 경로 지원: ROCm apt/rpm 저장소에서 바이너리 가져오기(이미지가 더 작고 빌드가 빠름) 또는 소스에서 컴파일(더 유연하지만 느림).
• Docker Compose를 사용하여 다중 컨테이너 설정을 관리하고 GPU 장치 플래그를 기억합니다.
• Docker Hub에서 사전 빌드된 rocm/rocm-terminal 이미지를 제공하여 빠른 시작을 지원합니다.
• 터미널 컨테이너에 비밀번호 없는 sudo를 가진 편리한 비루트 사용자 'rocm-user'를 포함하며, 필요 시 보안을 강화하는 지침을 제공합니다.
• FAQ는 누락된 apt 메타데이터, Docker 권한 오류, 컨테이너 안팎으로 파일 복사, 컨테이너를 이미지로 커밋하여 작업 보존과 같은 일반적인 문제를 다룹니다.