From 9fdc2f73a63e1f6fd64fdad06f37aef1f97b0adc Mon Sep 17 00:00:00 2001 From: Henk Date: Tue, 9 May 2023 20:59:10 +0200 Subject: [PATCH] ROCM Downgrade for stability --- environments/rocm.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/environments/rocm.yml b/environments/rocm.yml index 51b3e852..81e32a58 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -25,7 +25,7 @@ dependencies: - psutil - pip: - --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 - - torch==2.0.* + - torch==1.13.* - flask-cloudflared==0.0.10 - flask-ngrok - flask-cors