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