From 06d8e82e47a332c87f3260faa2034c563ac75d98 Mon Sep 17 00:00:00 2001 From: Max Kuhmichel Date: Thu, 29 Aug 2024 11:39:40 +0000 Subject: [PATCH] fix typo --- images/deepml/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/images/deepml/Dockerfile b/images/deepml/Dockerfile index 479f6e2..9416acd 100644 --- a/images/deepml/Dockerfile +++ b/images/deepml/Dockerfile @@ -57,7 +57,7 @@ ENV TF_FORCE_GPU_ALLOW_GROWTH true # Install PyTorch RUN pip install \ - torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 + torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 \ #torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117 \ && rm -rf /tmp/* $HOME/.cache