diff --git a/dockerlab/templates/workspace_pytorch_2_2/Dockerfile b/dockerlab/templates/workspace_pytorch_2_2/Dockerfile index ab35e57..ca6408a 100644 --- a/dockerlab/templates/workspace_pytorch_2_2/Dockerfile +++ b/dockerlab/templates/workspace_pytorch_2_2/Dockerfile @@ -7,5 +7,5 @@ FROM deepbase/dockerlab:workspace_cuda_12_1 # https://pytorch.org/get-started/previous-versions/ # https://github.com/pytorch/pytorch/wiki/PyTorch-Versions RUN pip install --no-cache-dir \ - torch==2.2.0 \ + torch==2.2.1 \ --extra-index-url https://download.pytorch.org/whl/cu118