Open alexander-sony opened 2 hours ago
maybe update readme with
git clone... && cd ...
docker build -f dockerfiles/cuda120/Dockerfile -t benchmarking-cv-models:cuda120 .
docker run --rm --ipc=host --ulimit memlock=-1 --gpus all benchmarking-cv-models:cu120 src.train --batch-size 32
testing minimal example from readme.md: