Update base docker image

isl-org / MiDaS

Code for robust monocular depth estimation described in "Ranftl et. al., Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer, TPAMI 2022"

MIT License

4.27k stars 599 forks source link

Update base docker image #211

Open IwfY opened 1 year ago

IwfY commented 1 year ago

Using nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04 adds CUDA support for newer graphic chips.
imutils popped up as a missing dependency for me.

figuernd commented 1 year ago

This might also fix the broken GPG key error that comes with the older image

figuernd commented 1 year ago

Confirmed that it does fix the NVIDIA public key rotation issue in older images, but I notice that the build fails trying to cd weights because that folder is never copied or created. Unsure how the original Dockerfile ever built without that step.