saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
transformers/docker/transformers-pytorch-amd-gpu
History
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