diff --git a/docker/Dockerfile b/docker/Dockerfile
index fcf8b2ff4..6fe58b15c 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,9 +1,9 @@
 ARG CUDA="11.3.0"
 ARG PYTORCH="1.10"
-ARG PYTHON="3.10"
+ARG PYTHON="3.8"
 
 # TODO: conda installs its own version of cuda
-FROM nvidia/cuda:${CUDA}-devel-ubuntu20.04
+FROM nvidia/cuda:${CUDA}-devel-ubuntu18.04
 
 ENV CUDA_PATH /usr/local/cuda
 ENV CUDA_ROOT /usr/local/cuda/bin