[Docker] pytorch ROCm gfx803 빌드

개발 2022년 12월 12일

없길래 직접 만들어왔어요

Docker Hub
docker run -it --privileged --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 8G --name <yourcontainername> pikokr/pytorch-gfx803:1.14.0 bash

rocm/pytorch:latest-base 기반이고요 python 3.10으로 빌드됐어요.

태그

파링

바보