coreweave / ml-containers

MIT License
19 stars 3 forks source link

fix(torch): Install missing runtime dependencies #38

Closed Eta0 closed 12 months ago

Eta0 commented 12 months ago

PyTorch & Torchvision Runtime Dependencies

This change installs runtime libraries that were previously missing from the final ml-containers/torch image:

  1. libpng16-16
  2. libjpeg-turbo8
  3. libnvjpeg-X-Y (for a given CUDA version X.Y)
  4. The latest libstdc++6

This change also switches the regular vim install to vim-tiny instead to save about 62 MiB, making up for any size increase.