Skip to content

Commit

Permalink
Merge pull request #41 from fhswf/DeepCowProductions-patch-1
Browse files Browse the repository at this point in the history
Update Dockerfile
  • Loading branch information
DeepCowProductions authored Nov 2, 2024
2 parents fae2106 + b49a7c0 commit 0415b3a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion images/deepml/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ RUN pip install \

# Install PyTorch
RUN pip install \
torch==2.4.0+cu124 torchvision==0.19.0+cu124 torchaudio==2.4.0+cu124 --index-url https://download.pytorch.org/whl/cu124
torch==2.4.0+cu124 torchvision==0.19.0+cu124 torchaudio==2.4.0+cu124 --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

Expand Down

0 comments on commit 0415b3a

Please sign in to comment.