Pyrit with CUDA as docker container using nvidia-docker

Uses pyrit as default entrypoint with GPU cores only.

Requirements:

• nvidia-smi
• nvidia-docker
• CUDA >= 9

Available CUDA versions:

• 11.4.1
• 11.4.0
• 11.3.1
• 11.3.0
• 11.2.2
• 11.2.1
• 11.2.0
• 11.1.1
• 11.0.3
• 10.2
• 10.1
• 10.0
• 9.2

Check your CUDA version

nvidia-smi | grep CUDA | sed 's/.*Version: //' | sed 's/ .*//'

Match your version with available ones

VERSION=11.4.1

Build image

docker build --build-arg CUDA_VERSION=$VERSION -t pyrit-cuda .

Run (as pyrit is the entrypoint, you can specify a command and parameters)

docker run --rm --gpus all pyrit-cuda