diff --git a/Dockerfile b/Dockerfile
index d9ce810..bfa150b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,13 +4,21 @@ ARG APT_MIRROR=https://mirrors.bfsu.edu.cn
 ARG ROOT_PASSWD=miao
 
 RUN \
+    env | grep -E -e '^NVIDIA' -e '^NV_' -e '^NVARCH' -e '^NCCL_' | \
+        sed -E 's/=(.*)/="\1"/' > \
+        /etc/profile.d/00-nvidia.sh && \
+    echo "PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:\${PATH}" >> /etc/profile.d/00-nvidia.sh && \
+    echo "LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64" >> /etc/profile.d/00-nvidia.sh && \
+    echo "LIBRARY_PATH=/usr/local/cuda/lib64/stubs" >> /etc/profile.d/00-nvidia.sh && \
+    sed -i 's/^/export /' /etc/profile.d/00-nvidia.sh && \
+    \
     sed -i "s#http://archive.ubuntu.com#${APT_MIRROR}#g" /etc/apt/sources.list && \
     sed -i "s#http://security.ubuntu.com#${APT_MIRROR}#g" /etc/apt/sources.list && \
     \
     echo root:${ROOT_PASSWD} | chpasswd && \
     \
     apt update && \
-    apt install --yes ssh tini && \
+    apt install --yes ssh tini neovim git && \
     apt clean && \
     rm -rf /var/lib/apt/lists/* && \
     \
diff --git a/compose.yml b/compose.yml
index faab0f6..3925a68 100644
--- a/compose.yml
+++ b/compose.yml
@@ -1,4 +1,3 @@
-#--rm --device nvidia.com/gpu=all -it docker.io/nvidia/cuda:11.8.0-devel-ubuntu22.04 bash
 version: "3.9"
 name: "mcuda"
 services: