diff --git a/Dockerfile b/Dockerfile
index 723f42f..97c5866 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/cuda:11.7.0-cudnn8-devel-ubuntu18.04
+FROM nvcr.io/nvidia/cuda:11.7.0-cudnn8-devel-ubuntu20.04
 
 RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
 RUN echo "en_US.UTF-8 UTF-8" >> /etc/locale.gen