Docker Instructions

This directory contains the Dockerfile for setting up the FireANTs environment with CUDA support.

Prerequisites

First, clone the repository:

git clone https://github.com/rohitrango/fireants
cd fireants

Then, from the root directory of the project, run:

docker buildx build -t fireants -f docker/Dockerfile .

This will create a Docker image named fireants with all the necessary dependencies installed.

Note: If your system has CUDA 11.8, you can use the Dockerfile-cu118 dockerfile instead.

To run the container with GPU support:

docker run --gpus all --shm-size=40g -it -v /data:/data fireants

To mount your local code directory for development:

docker run --gpus all --shm-size=40g -it -v $(pwd):/fireants -v /data:/data fireants

You may need to rebuild the fused_ops extension after mounting the code directory if you are making changes to the fused_ops kernels.