Refactor Dockerfile to include CUDA environment variables in /etc/pro…

…file
labring-actions · Dec 11, 2024 · df331ef · df331ef
1 parent cde85f4
commit df331ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/OS/ubuntu-cuda/24.04/Dockerfile b/OS/ubuntu-cuda/24.04/Dockerfile
@@ -37,6 +37,8 @@ RUN chmod +x /usr/start/startup.sh && \
     useradd -m -s /bin/bash devbox && \
     usermod -aG sudo devbox && \
     echo 'devbox ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers && \
+    echo 'export PATH=/usr/local/cuda/bin:$PATH' >>  /etc/profile && \
+    echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> /etc/profile && \
     rm -rf /tmp/* && \
     mkdir -p /home/devbox/.ssh && \
     chown -R devbox:devbox /home/devbox/.ssh && \

diff --git a/OS/ubuntu/24.04/Dockerfile b/OS/ubuntu/24.04/Dockerfile
@@ -36,8 +36,6 @@ RUN chmod +x /usr/start/startup.sh && \
     useradd -m -s /bin/bash devbox && \
     usermod -aG sudo devbox && \
     echo 'devbox ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers && \
-    echo 'export PATH=/usr/local/cuda/bin:$PATH' >>  /etc/profile && \
-    echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> /etc/profile && \
     rm -rf /tmp/* && \
     mkdir -p /home/devbox/.ssh && \
     chown -R devbox:devbox /home/devbox/.ssh && \