Visit NVIDIA CUDA compatibility matrix and install the latest driver as shown in utils/install_nvidia_430_18.04.sh
.
Check whether the driver is properly installed and its version:
nvidia-smi
To install Docker and NVIDIA-docker2 execute utils/install_docker.sh
.
If the installation succeeds, the list of GPUs available within a docker container will be shown.
Set a password for Jupyter in run/config/jupyter_notebook_config.py
.
See run/README.md or just execute:
cd run
./run.sh
Specify FULL_IMAGE_NAME
in action.
Set DOCKER_HUB_TOKEN
and DOCKER_HUB_USER
secrets to allow pushing the image to Docker Hub.