Sign inSign up

hardandheavy/transformers-rocm

By hardandheavy

Updated 8 months ago

https://github.com/HardAndHeavy/transformers-rocm-docker

Image
Machine learning & AI
1

8.3K

hardandheavy/transformers-rocm repository overview

Transformers on GPU AMD Radeon in Docker

Basic configuration of the docker container hardandheavy/transformers-rocm for working with llama models on GPU from Radeon.

The source code for the container is located at https://github.com/HardAndHeavy/transformers-rocm-docker.

Tag summary

Content type

Image

Digest

sha256:8d4782887

Size

11.2 GB

Last updated

8 months ago

docker pull hardandheavy/transformers-rocm