Add libnvjpeg-dev package as fallback (#1294)

* Add libnvjpeg-dev package as fallback * Move libnvjpeg and libnvjpeg-dev to required packages * Update conda/pytorch-cuda/meta.yaml --------- Co-authored-by: Nikita Shulga <[email protected]>
pytorch · Feb 7, 2023 · 177ba26 · 177ba26
1 parent 16fb06e
commit 177ba26
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml
@@ -41,6 +41,9 @@ requirements:
   run:
     - cuda-libraries {{ cuda_constraints }}
     - cuda-nvtx {{ cuda_constraints }}
+    - libnvjpeg {{ libnvjpeg_constraints }}
+    # libnvjpeg-dev required see: https://github.com/pytorch/vision/issues/7185#issuecomment-1420002413
+    - libnvjpeg-dev {{ libnvjpeg_constraints }}
   run_constrained:
     - cuda-cudart {{ cuda_constraints }}
     - cuda-cupti {{ cuda_constraints }}
@@ -51,7 +54,6 @@ requirements:
     - libcusolver {{ libcusolver_constraints }}
     - libcusparse {{ libcusparse_constraints }}
     - libnpp {{ libnpp_constraints }}
-    - libnvjpeg {{ libnvjpeg_constraints }}
 test:
   commands:
     - echo "pytorch-cuda metapackage is created."