Skip to content

Commit

Permalink
Add dev packages for windows pytorch-cuda dependencies (#1295)
Browse files Browse the repository at this point in the history
* Add dev packages for windows dependencies

* Adding architecture dependent builds

* Add notes around windows

* fix typo

* Bumping version to v3
  • Loading branch information
atalman authored Feb 8, 2023
1 parent 9a931c5 commit da6dd4c
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 5 deletions.
3 changes: 3 additions & 0 deletions conda/pytorch-cuda/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,6 @@ version:
- 11.6
- 11.7
- 11.8
target_platform:
- win-64
- linux-64
21 changes: 16 additions & 5 deletions conda/pytorch-cuda/meta.yaml
Original file line number Diff line number Diff line change
@@ -1,4 +1,10 @@
# Package to manage cuda version in PyTorch
# Package to manage cuda version in PyTorch.
#
# Windows anaconda packages are packaged differently,
# All dlls are kept within *-dev packages hence we need
# include the dev packages for Windows see:
# https://github.com/pytorch/vision/issues/7185#issuecomment-1420002413
#
# Please note: Build number should be advanced with
# every deployment. After the deployment to production
# use following links to validate the correctness of
Expand Down Expand Up @@ -36,15 +42,11 @@ package:
version: {{ version }}
build:
number: {{ build }}
noarch: generic
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 }}
- cuda-nvrtc {{ cuda_constraints }}
Expand All @@ -54,6 +56,15 @@ requirements:
- libcusolver {{ libcusolver_constraints }}
- libcusparse {{ libcusparse_constraints }}
- libnpp {{ libnpp_constraints }}
- cuda-libraries-dev {{ cuda_constraints }} # [win64]
- libnvjpeg-dev {{ libnvjpeg_constraints }} # [win64]
- cuda-cudart-dev {{ cuda_constraints }} # [win64]
- cuda-nvrtc-dev {{ cuda_constraints }} # [win64]
- libcufft-dev {{ libcufft_constraints }} # [win64]
- libcublas-dev {{ libcublas_constraints }} # [win64]
- libcusolver-dev {{ libcusolver_constraints }} # [win64]
- libcusparse-dev {{ libcusparse_constraints }} # [win64]
- libnpp-dev {{ libnpp_constraints }} # [win64]
test:
commands:
- echo "pytorch-cuda metapackage is created."
Expand Down

0 comments on commit da6dd4c

Please sign in to comment.