diff --git a/Dockerfile b/Dockerfile
index 3d3b26a..9896245 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+FROM ubuntu:latest
 
 WORKDIR /subgen