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