Skip to content

Latest commit

 

History

History
24 lines (16 loc) · 893 Bytes

README.md

File metadata and controls

24 lines (16 loc) · 893 Bytes

rocm-pytorch-gfx803-docker

A Docker image based on rocm/pytorch with support for gfx803(Polaris 20-21 (XT/PRO/XL); RX580; RX570; RX560) and Python 3.8

For ROCM 5.3.0

After creating container, you will be logged as sduser with activated python3.8 environment.

Supports docker and podman.

torch, torchvision and tensorflow-rocm packages are downloaded from https://github.com/xuhuisheng/rocm-gfx803

Every downloaded package should be in /packages.

Building image

podman build -t rocm-pytorch-gfx803 ./rocm-pytorch-gfx803-docker

Creating container

podman run -it --device=/dev/kfd --device=/dev/dri --net=host localhost/rocm-pytorch-gfx803

Not required, but recommended is -v /host:/container parameter. It is easier to work when you have easy access to files already in your system. /host can be any path on host and /container any path on container.