This project generates a docker image that enables rapid development with the fast.ai v1 library with Modern AMD graphics cards through the Rocm stack.
This docker image contains:
- Jupyter Notebook
- Pytorch
- Fast.ai
- Rocm Utilities
- Python 3.6
Usage
sudo docker run -it -v $HOME:/data -p 1337:1337 --privileged --rm --device=/dev/kfd --device=/dev/dri --group-add video briangorman/fastai_rocm
Your home directory will be exposed in /data. The docker container will output a link to access the jupyter notebook