Append cuda binary path.

Remove cuda installer to reduce image size.
sailfish009 · Sep 12, 2021 · 92115dd · 92115dd
1 parent cb11811
commit 92115dd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
@@ -1,4 +1,6 @@
 FROM ghcr.io/deepmodeling/abacus-development-kit:gnu
 
 RUN wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda_11.4.2_470.57.02_linux.run --no-check-certificate --quiet \
- && sh cuda_11.4.2_470.57.02_linux.run --toolkit --silent
+ && sh cuda_11.4.2_470.57.02_linux.run --toolkit --silent \
+ && rm cuda_11.4.2_470.57.02_linux.run
+ENV PATH=/usr/local/cuda/bin:$PATH