Update Dockerfile CUDA (#2682)

* Update Dockerfile CUDA

* Apply suggestions from code review
This commit is contained in:
Patrick von Platen 2023-03-15 18:02:56 +01:00 committed by GitHub
parent 4553c29d92
commit 9d1341d69b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions

View File

@ -27,7 +27,6 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
torch \
torchvision \
torchaudio \
--extra-index-url https://download.pytorch.org/whl/cu117 && \
python3 -m pip install --no-cache-dir \
accelerate \
datasets \
@ -40,4 +39,4 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip && \
tensorboard \
transformers
CMD ["/bin/bash"]
CMD ["/bin/bash"]