From e41f3b301329b052fea8d530d490e7cbe0d8d8c8 Mon Sep 17 00:00:00 2001
From: georgeyiasemis <georgeyiasemis@hotmail.com>
Date: Sun, 23 Jun 2024 12:52:29 +0200
Subject: [PATCH] Revert back

---
 docker/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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