tensorpix / benchmarking-cv-models

Benchmark computer vision ML models in 3 minutes
MIT License
16 stars 0 forks source link

docker error for minimal example #19

Open alexander-sony opened 2 hours ago

alexander-sony commented 2 hours ago

testing minimal example from readme.md:

docker run --rm --ipc=host --ulimit memlock=-1 --gpus all ghcr.io/tensorpix/benchmarking-cv-models src.train --batch-size 32
Unable to find image 'ghcr.io/tensorpix/benchmarking-cv-models:latest' locally
docker: Error response from daemon: Head "https://ghcr.io/v2/tensorpix/benchmarking-cv-models/manifests/latest": denied: denied.
alexander-sony commented 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