Skip to content

Latest commit

 

History

History
38 lines (31 loc) · 815 Bytes

README.md

File metadata and controls

38 lines (31 loc) · 815 Bytes

Catalyst Docker

Docker Pulls

Base version

Base docker has Catalyst and all needed requirements.

make docker

With FP16

make docker-fp16

Developer version

The developer version contains packages for building docs, for checking the code style. And does not contain Catalyst itself.

make docker-dev

With FP16

make docker-dev-fp16

How to use

export GPUS=...
export LOGDIR=...
docker run -it --rm --runtime=nvidia \
   -v $(pwd):/workspace -v $LOGDIR:/logdir/ \
   -e "CUDA_VISIBLE_DEVICES=${GPUS}" \
   catalyst-base catalyst-dl run \
   --config=./configs/train.yml --logdir=/logdir