diff --git a/docker/initialize.sh b/docker/initialize.sh
index ad38019..719492e 100755
--- a/docker/initialize.sh
+++ b/docker/initialize.sh
@@ -29,8 +29,6 @@ cd ..
 cd ..
 
 
-#sudo apt-get -y install cuda
-#sudo apt -y install nvidia-cuda-toolkit
 
 python3.10 -m pip install plyfile tqdm
 python3.10 -m pip install  https://huggingface.co/camenduru/gaussian-splatting/resolve/main/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl
@@ -39,4 +37,7 @@ python3.10 -m pip install torchvision
 
 ln -s docker/run.sh ./run.sh
 
+#sudo apt-get -y install cuda
+sudo apt -y install nvidia-cuda-toolkit
+
 exit 0