diff --git a/python3-pytorch-pip/install.bash b/python3-pytorch-pip/install.bash index f38342457..ae5244505 100644 --- a/python3-pytorch-pip/install.bash +++ b/python3-pytorch-pip/install.bash @@ -5,6 +5,6 @@ then tue-install-pip torch tue-install-pip torchvision else - tue-install-pip "torch -i https://download.pytorch.org/whl/cpu" - tue-install-pip "torchvision -i https://download.pytorch.org/whl/cpu" + tue-install-pip "torch==1.13 -i https://download.pytorch.org/whl/cpu" + tue-install-pip "torchvision==0.14 -i https://download.pytorch.org/whl/cpu" fi