tensorpix / benchmarking-cv-models

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

Set ubuntu22.04 as base image; Bump torch and lighting #15

Closed bfreskura closed 7 months ago

bfreskura commented 7 months ago

Torch actually comes with cuda and cudnn included so we don't need it in the base image.

Reduces image size by ~4GB

https://discuss.pytorch.org/t/does-pytorch-use-the-cudatoolkit-in-docker-or-the-system/140303