Dockerfiles update (#894)

Author: dxqb

resources/docker/RunPod-NVIDIA-CLI.Dockerfile

@@ -3,8 +3,8 @@
 #    docker tag <image-name> <dockerhub-username>/<repository-name>:<tag>
 #    docker push <dockerhub-username>/<repository-name>:<tag>
 
-FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04
-#the base image is barely used, pytorch 2.4 is the wrong version. However, by using
+FROM runpod/pytorch:2.8.0-py3.11-cuda12.8.1-cudnn-devel-ubuntu22.04
+#the base image is barely used, pytorch is the wrong version. However, by using
 #a base image that is popular on RunPod, the base image likely is already available
 #in the image cache of a pod, and no download is necessary
 

resources/docker/Vast-NVIDIA-CLI.Dockerfile

@@ -0,0 +1,30 @@
+#To build, run
+#    docker build -t <image-name> . -f Vast-NVIDIA-CLI.Dockerfile
+#    docker tag <image-name> <dockerhub-username>/<repository-name>:<tag>
+#    docker push <dockerhub-username>/<repository-name>:<tag>
+
+FROM vastai/pytorch:cuda-12.8.1-auto
+
+WORKDIR /
+RUN git clone https://github.com/Nerogar/OneTrainer
+RUN cd OneTrainer \
+ && export OT_PLATFORM_REQUIREMENTS=requirements-cuda.txt \
+ && export OT_LAZY_UPDATES=true \
+ && export OT_PYTHON_CMD=/venv/main/bin/python \
+ && ./install.sh \
+ && pip cache purge \
+ && rm -r ~/.cache/pip
+RUN apt-get update --yes \
+ && apt-get install --yes --no-install-recommends \
+      joe \
+	  less \
+	  gh \
+	  iputils-ping \
+	  nano \
+ && apt-get autoremove -y \
+ && apt-get clean \
+ && rm -rf /var/lib/apt/lists/*
+RUN pip install nvitop \
+ && pip cache purge \
+ && rm -rf ~/.cache/pip
+RUN ln -snf /OneTrainer /workspace/OneTrainer