diff --git a/environments/rocm.yml b/environments/rocm.yml index b79615c2..dcd76b05 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -21,7 +21,7 @@ dependencies: - psutil - pip: - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - - torch==1.11.* + - torch==1.12.1+rocm5.1.1 - flask-cloudflared - flask-ngrok - lupa==1.10