ROCm torch version pin

2023-01-08 17:10:59 +01:00 · 2023-01-08 17:10:59 +01:00 · 857476ef6b
parent 7fc5c46c1d
commit 857476ef6b
1 changed files with 1 additions and 2 deletions
--- 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