diff --git a/environments/rocm.yml b/environments/rocm.yml index 9679d036..ad6ea39a 100644 --- a/environments/rocm.yml +++ b/environments/rocm.yml @@ -22,8 +22,8 @@ dependencies: - Pillow - psutil - pip: - - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 - - torch==1.12.1+rocm5.1.1 + - --extra-index-url https://download.pytorch.org/whl/rocm5.2 + - torch==1.13.1+rocm5.2 - flask-cloudflared==0.0.10 - flask-ngrok - flask-cors