search

mapbox / robosat

Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds
MIT License
2.02k stars 383 forks source link

latest docker gpu problem #192

Closed dorbodwolf closed 4 years ago

dorbodwolf commented 4 years ago

reproduce the error:

deekongone# docker run --runtime=nvidia -it --rm -v $PWD:/data --ipc=host mapbox/robosat:latest-gpu train --mode
l /data/model.toml --dataset /data/dataset.toml --workers 4

error as below:

docker: Error response from daemon: OCI runtime create failed: container_linux.go:348: starting container process caused "exec: \"train\": executable file not found in $PATH": unknown. deekongone#

daniel-j-h commented 4 years ago

Can not reproduce; the following works calling rs train with the help flag:

djh@rig ~> docker run --runtime=nvidia -it --rm mapbox/robosat:latest-gpu train --help
usage: ./rs train [-h] --model MODEL --dataset DATASET
                  [--checkpoint CHECKPOINT] [--resume RESUME]
                  [--workers WORKERS]

optional arguments:
  -h, --help            show this help message and exit
  --model MODEL         path to model configuration file (default: None)
  --dataset DATASET     path to dataset configuration file (default: None)
  --checkpoint CHECKPOINT
                        path to a model checkpoint (to retrain) (default:
                        None)
  --resume RESUME       resume training or fine-tuning (if checkpoint)
                        (default: False)
  --workers WORKERS     number of workers pre-processing images (default: 0)
dorbodwolf commented 4 years ago

@daniel-j-h yes, now its works, I dont know why..thanks I will close this issue.