From 177ba2677ff6fcb728bfdf530d264bdfa25ce4ea Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Tue, 7 Feb 2023 11:59:08 -0500 Subject: [PATCH] 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 --- conda/pytorch-cuda/meta.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml index 7fa03466f..25e9a5a55 100644 --- 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."