Make faster docker build

[vwxyzjn]

This stuff never changes. We should just use a prebuilt one or build one ourselves to save time building. Many times I am just waiting for the miniconda install.

open-instruct/Dockerfile

Lines 1 to 66 in 5834277

	ARG CUDA
	ARG DIST
	ARG TARGET
	FROM --platform=linux/amd64 nvidia/cuda:${CUDA}-${TARGET}-${DIST}
	ARG DEBIAN_FRONTEND="noninteractive"
	ENV TZ="America/Los_Angeles"
	# Install base tools.
	RUN apt-get update && apt-get install -y \
	build-essential \
	curl \
	git \
	jq \
	language-pack-en \
	make \
	sudo \
	unzip \
	vim \
	wget \
	parallel \
	iputils-ping \
	tmux
	# This ensures the dynamic linker (or NVIDIA's container runtime, I'm not sure)
	# puts the right NVIDIA things in the right place (that THOR requires).
	ENV NVIDIA_DRIVER_CAPABILITIES=graphics,utility,compute
	# Install conda. We give anyone in the users group the ability to run
	# conda commands and install packages in the base (default) environment.
	# Things installed into the default environment won't persist, but we prefer
	# convenience in this case and try to make sure the user is aware of this
	# with a message that's printed when the session starts.
	RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.1.0-1-Linux-x86_64.sh \
	&& echo "32d73e1bc33fda089d7cd9ef4c1be542616bd8e437d1f77afeeaf7afdb019787 Miniconda3-py310_23.1.0-1-Linux-x86_64.sh" \
	| sha256sum --check \
	&& bash Miniconda3-py310_23.1.0-1-Linux-x86_64.sh -b -p /opt/miniconda3 \
	&& rm Miniconda3-py310_23.1.0-1-Linux-x86_64.sh
	ENV PATH=/opt/miniconda3/bin:/opt/miniconda3/condabin:$PATH
	ENV LD_LIBRARY_PATH=/usr/local/cuda/lib:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
	# Install a few additional utilities via pip
	RUN /opt/miniconda3/bin/pip install --no-cache-dir \
	gpustat \
	jupyter \
	beaker-gantry \
	oocmap
	# Ensure users can modify their container environment.
	RUN echo '%users ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers
	# Make the base image friendlier for interactive workloads. This makes things like the man command
	# work.
	RUN yes | unminimize
	# Install MLNX OFED user-space drivers
	# See https://docs.nvidia.com/networking/pages/releaseview.action?pageId=15049785#Howto:DeployRDMAacceleratedDockercontaineroverInfiniBandfabric.-Dockerfile
	ENV MOFED_VER 5.8-1.1.2.1
	ENV OS_VER ubuntu20.04
	ENV PLATFORM x86_64
	RUN wget --quiet https://content.mellanox.com/ofed/MLNX_OFED-${MOFED_VER}/MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz && \
	tar -xvf MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz && \
	MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}/mlnxofedinstall --basic --user-space-only --without-fw-update -q && \
	rm -rf MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM} && \
	rm MLNX_OFED_LINUX-${MOFED_VER}-${OS_VER}-${PLATFORM}.tgz