Skip to content
This repository has been archived by the owner on Jun 28, 2024. It is now read-only.

evshiron/rocm_lab

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

ROCm LAB

Experiments to see the potential of RX 7000 series.

IMPORTANT!!!

You can now install the official torch nightly build for ROCm 5.5+ with:

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.5

# or this more performant ones if you are using rocm 5.6
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.6

For the official tensorflow nightly build, see here.

Motivations

As we know, ROCm 5.5.0 was released on May 2nd, 2023. After waiting for several days, we discovered that all official Docker images did not include support for the RX 7000 series (a.k.a. gfx1100), which are currently the best-performing and most suitable consumer-grade GPUs for the AI field under the AMD brand. As we aim to tap into the potential of RX 7000 series GPUs as soon as possible:

ROCm LAB will focus on conducting proof of concept and delivering prebuilt wheels specifically for the RX 7000 series until official support is provided. The goal of these efforts is to demonstrate the viability and effectiveness of using these GPUs for AI applications and to provide a basis for further development.

Additionally, the website Are we gfx1100 yet? will serve as a platform for showcasing the latest proof of concept developments in the AI field using the RX 7000 series.

Prebuilt wheels

Prebuilt wheels are built by GitHub Actions and can now be found in GitHub Releases.

It's worth noting that these wheels are built using GitHub's ubuntu-latest runner, which is Ubuntu 22.04 right now. There might be dynamic linking issues when used in other systems. If it does, consider building a Docker container instead.

How to use

Simple download the wheel you want and install it with:

# recommended: activate venv
source venv/bin/activate

# download the wheel
curl -L -O https://github.com/evshiron/rocm_lab/releases/download/v1.14.514/torch-2.0.1+gite19229c-cp310-cp310-linux_x86_64.whl

# install the wheel
pip install torch-2.0.1+gite19229c-cp310-cp310-linux_x86_64.whl

Prebuilt Docker images

These Docker images are mainly proofs of concepts and will not be updated frequently.

It's recommended to use the wheels above directly, or build your own Docker images with these wheels if you like.

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 -e HSA_OVERRIDE_GFX_VERSION=11.0.0 --name rocm5.5-automatic ghcr.io/evshiron/rocm_lab:rocm5.5-automatic

Are we gfx1100 yet?

Credits

A large portion of the content in this repository comes from the internet. My main work is to collect, experiment, and organize this information.

I would like to express my gratitude to other developers for their contributions, as well as those who have discussed with me and provided me with assistance.

If you find my work helpful, please consider giving this repository a star. Your recognition will be my motivation.

License

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