- Nvidia drivers >= 470, CUDA >= 11.4
- Docker, nvidia-docker
The provided Dockerfile is supplied to build an image with CUDA support and cuDNN.
Also you can install the necessary python packages via requirements.txt
- Clone the repo.
- Download and extract dataset to the
data/folder. sudo make allto build a docker image and create a container. Orsudo make all GPUS=device=0 CPUS=10if you want to specify gpu devices and limit CPU-resources.