feat: replace latest tag on default sshbox image (#29)

sshbox/Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:latest
+FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-devel
 
 # Install SSH server
 RUN apt-get update && \
@@ -18,8 +18,9 @@ RUN echo "TODO" > /home/ubuntu/.ssh/authorized_keys
 # RUN chown -R ubuntu:ubuntu /home/ubuntu/.ssh
 # RUN chmod 600 /home/ubuntu/.ssh/authorized_keys
 
-# Permit root login via SSH
-# RUN sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
+# Add CUDA binaries to PATH and CUDA libraries to LD_LIBRARY_PATH
+RUN echo 'export PATH="/usr/local/cuda/bin:${PATH}"' >> /etc/bash.bashrc
+RUN echo 'export LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"' >> /etc/bash.bashrc
 
 # switch port to 2222
 RUN sed -i 's/#Port 22/Port 2222/' /etc/ssh/sshd_config