diff --git a/environments/h100env_requirements.txt b/environments/h100env_requirements.txt index e44fbda..8d462fd 100644 --- a/environments/h100env_requirements.txt +++ b/environments/h100env_requirements.txt @@ -65,7 +65,7 @@ setproctitle==1.3.4 six==1.17.0 smmap==5.0.2 sympy==1.13.1 -torch==2.5.1+cu121 +torch==2.12.0 torchaudio==2.5.1+cu121 torchmetrics==1.6.1 torchvision==0.20.1+cu121