transformers/docker/transformers-pytorch-amd-gpu
ivarflakstad bc6ae0d55e
Update AMD docker image (rocm 6.1) (#35259)
* Use rocm 6.3 as base amd image and add nvidia-ml-py to exclude list

* Align rocm base image with torch wheels @6.1. Seems like the most stable combo
2024-12-13 15:41:03 +01:00
..
Dockerfile Update AMD docker image (rocm 6.1) (#35259) 2024-12-13 15:41:03 +01:00