diff --git a/nasp-ubuntu/Dockerfile b/nasp-ubuntu/Dockerfile
index e4d4b0b..4d69ee1 100644
--- a/nasp-ubuntu/Dockerfile
+++ b/nasp-ubuntu/Dockerfile
@@ -1,4 +1,4 @@
-FROM docker.io/ubuntu:22.04
+FROM docker.io/nvidia/cuda:11.3.1-devel-ubuntu20.04
 
 # Install dependencies
 RUN ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
@@ -17,16 +17,6 @@ RUN cd /tmp && wget http://192.168.16.118/MLNX_OFED_LINUX-5.8-2.0.3.0-ubuntu22.0
   && ./mlnxofedinstall --skip-unsupported-devices-check -q \
   && rm -rf /tmp/MLNX_OFED_LINUX-5.8-*
 
-# cuda
-RUN apt-get update -y && apt-get install -yq linux-headers-$(uname -r) \
-  && apt-key del 7fa2af80 && cd /tmp && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb \
-  && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get update -y \
-  && DEBIAN_FRONTEND=noninteractive apt-get install -yq --no-install-recommends cuda-11-3 cuda-compat-12-0
-
-ENV PATH=/usr/local/cuda/bin:$PATH \
-  LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
-  CUDA_HOME=/usr/local/cuda
-
 # startup
 COPY startup.sh /etc/startup.sh
 ENTRYPOINT ["/etc/startup.sh"]