julia.def

Bootstrap: docker
From: julia:1.1.0

%environment
	# If you want the julia packages to be installed
	# in a different place than ~/.julia, modify
	# the variable bellow
	# export JULIA_DEPOT_PATH=/some/place/.julia
	export JULIA_PROJECT=@.
	export PYTHON=/usr/lib/x86_64-linux-gnu/
  export GKSwstype=svg

	export CUDA_HOME=/usr/local/cuda
	CUDA_LIB=$CUDA_HOME/lib64
	CUDA_INCLUDE=$CUDA_HOME/include
	CUDA_BIN=$CUDA_HOME/bin
	export LD_LIBRARY_PATH=$CUDA_LIB:/usr/local/lib
	export PATH=$CUDA_BIN:$PATH

%runscript
  export JULIA_NUM_THREADS="$(( `nproc` / 2 ))"
  exec -B /mnt/storage,/usr/local/cuda /usr/local/julia/bin/julia -O 3 "$@"

%post
	apt update
	apt install -y wget ssh vim gnupg apt-utils gcc gdb strace
	# NVIDIA
	wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
	dpkg -i cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
	apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
	wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
	dpkg -i nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
	rm *-repo-*deb
	apt clean
	apt-get update && apt-get -y install libcublas10 libcudnn7

	apt install -y python3 python3-pip \
		ffmpeg zlib1g hdf5-tools libcairo2 pdf2svg imagemagick \
		libczmq-dev cmake xorg-dev dvipng libxt6 libxrender1 \
		mesa-utils libxrandr-dev libxinerama-dev libxcursor-dev \
		libgl1-mesa-glx libqt5widgets5 gettext gettext-base

	# VirtualGL
	wget https://sourceforge.net/projects/virtualgl/files/2.6.1/virtualgl_2.6.1_amd64.deb
	dpkg -i virtualgl_2.6.1_amd64.deb
	rm virtualgl_2.6.1_amd64.deb

	# JuptyerLab
	pip3 install jupyterlab

  # LaTeX
  apt install -y texlive-latex-base \
		texlive-latex-recommended \
    texlive-luatex \
		texlive-latex-extra \
    texlive-generic-extra \
    texlive-science