search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.35k stars 6.97k forks source link

Building libtorchvision.so with nvJPEG #8751

Open srasmussenvl opened 1 week ago

srasmussenvl commented 1 week ago

🐛 Describe the bug

When I build libtorchvision.so with cmake, it ends up without nvJPEG. How should I modify CMakeLists.txt in order to get this to work?

Hope you can help me :) /Soren

Versions

0.20.0

NicolasHug commented 1 week ago

Hi @srasmussenvl

Sorry, we haven't enabled nvjpeg support on the Cmake-based build just yet. Unfortunately I cannot treat that as high-priority for me to do right now, but I'm happy to consider and review a PR from the community!