storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-15 00:03:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

update docker image to fix RTX3090 issue (needs a lot of cleanup)

Browse Source
This commit is contained in:
Brandon Thomas
2022-07-14 16:35:24 -04:00
parent 54e63f5b71
commit dfa6354a4e
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+3
View File
@@ -182,3 +182,6 @@ RUN python3 --version
# TODO: Use venv to do this instead.
RUN pip3 install gdown
# NB(echelon,2022-07-14): Run a newer version of pytorch which will support RTX3090.
# Fixes bug, "NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation."
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113