Bump torch ver and docs

2024-03-15 21:43:57 +09:00
parent 14c6805337
commit ab4020545e
3 changed files with 4 additions and 3 deletions
--- a/Dockerfile.train
+++ b/Dockerfile.train
@@ -90,7 +90,7 @@ ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

 # Based on https://pytorch.org/get-started/locally/

-RUN $PIP_INSTALL torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
+RUN $PIP_INSTALL torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118


 RUN $PIP_INSTALL jupyterlab