From dfa6354a4e62c365d6bb40ff9d5e304f0d47465f Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Thu, 14 Jul 2022 16:35:24 -0400 Subject: [PATCH] update docker image to fix RTX3090 issue (needs a lot of cleanup) --- Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Dockerfile b/Dockerfile index 1644058..c4d450b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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