Dockerized Environment

This page describes how to use the Dockerized environment

NVIDIA-Docker Build

cd <path_to_SIT_FUSE>/SIT_FUSE/hpc/docker

./build.sh latest

See Docker's documentation for further details on running containers:

What is a container?Docker Documentation

NVIDIA Container Toolkit + Docker

The nvidia-docker wrapper has been deprecated in favor of the NVIDIA Container Tookit. You can read more about installation and setup here:

Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkitdocs.nvidia.com

Once initial setup is complete, build the image with this command

docker build -t sit_fuse:latest .
#In same directory as above example

To run the container, you will need a few special arguments (namely --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=nvidia.com/gpu=all). The proper values to use, for your specific case are detailed in the NVIDIA Container Toolkit link above.

docker run -it -v /data/:/data/ \
--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=nvidia.com/gpu=all sit_fuse

The image has SIT_FUSE installed in the /app director