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