r3dlobst3r / rocm_docker

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

badge-torch-gfx1100 badge-tensorflow-gfx1100 badge-torchvision-gfx1100 badge-torchaudio-gfx1100 rocm_lab-downloads

ROCm Docker

Experiments to see the potential of AMD GPUs.

How to use

# add environment variables or volumes for your need
docker run -ti --net=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 8G --name automatic ghcr.io/r3dlobst3r/rocm_docker:automatic

License

This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.

About

License:Other


Languages

Language:Shell 100.0%