From 857476ef6b22df42a3e275948ff627fa8cebfa09 Mon Sep 17 00:00:00 2001 From: Henk Date: Sun, 8 Jan 2023 17:10:59 +0100 Subject: [PATCH] ROCm torch version pin --- environments/rocm.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/environments/rocm.yml b/environments/rocm.yml index f47dd6c1..b79615c2 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -21,8 +21,7 @@ dependencies: - psutil - pip: - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - - torch - - torchvision + - torch==1.11.* - flask-cloudflared - flask-ngrok - lupa==1.10