From 702f59b2dbd458ccc9426cee0226740870a62b36 Mon Sep 17 00:00:00 2001 From: Henk Date: Tue, 9 May 2023 22:10:01 +0200 Subject: [PATCH] Downgrade ROCM properly --- environments/rocm.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/environments/rocm.yml b/environments/rocm.yml index 81e32a58..a33a8f96 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -24,8 +24,8 @@ dependencies: - Pillow - psutil - pip: - - --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 - - torch==1.13.* + - --extra-index-url https://download.pytorch.org/whl/rocm5.2 + - torch==1.13.1+rocm5.2 - flask-cloudflared==0.0.10 - flask-ngrok - flask-cors