Skip to content

Commit da6dd4c

Browse files
authored
Add dev packages for windows pytorch-cuda dependencies (#1295)
* Add dev packages for windows dependencies * Adding architecture dependent builds * Add notes around windows * fix typo * Bumping version to v3
1 parent 9a931c5 commit da6dd4c

File tree

2 files changed

+19
-5
lines changed

2 files changed

+19
-5
lines changed

conda/pytorch-cuda/conda_build_config.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,3 +2,6 @@ version:
22
- 11.6
33
- 11.7
44
- 11.8
5+
target_platform:
6+
- win-64
7+
- linux-64

conda/pytorch-cuda/meta.yaml

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,10 @@
1-
# Package to manage cuda version in PyTorch
1+
# Package to manage cuda version in PyTorch.
2+
#
3+
# Windows anaconda packages are packaged differently,
4+
# All dlls are kept within *-dev packages hence we need
5+
# include the dev packages for Windows see:
6+
# https://github.com/pytorch/vision/issues/7185#issuecomment-1420002413
7+
#
28
# Please note: Build number should be advanced with
39
# every deployment. After the deployment to production
410
# use following links to validate the correctness of
@@ -36,15 +42,11 @@ package:
3642
version: {{ version }}
3743
build:
3844
number: {{ build }}
39-
noarch: generic
4045
requirements:
4146
run:
4247
- cuda-libraries {{ cuda_constraints }}
4348
- cuda-nvtx {{ cuda_constraints }}
4449
- libnvjpeg {{ libnvjpeg_constraints }}
45-
# libnvjpeg-dev required see: https://github.com/pytorch/vision/issues/7185#issuecomment-1420002413
46-
- libnvjpeg-dev {{ libnvjpeg_constraints }}
47-
run_constrained:
4850
- cuda-cudart {{ cuda_constraints }}
4951
- cuda-cupti {{ cuda_constraints }}
5052
- cuda-nvrtc {{ cuda_constraints }}
@@ -54,6 +56,15 @@ requirements:
5456
- libcusolver {{ libcusolver_constraints }}
5557
- libcusparse {{ libcusparse_constraints }}
5658
- libnpp {{ libnpp_constraints }}
59+
- cuda-libraries-dev {{ cuda_constraints }} # [win64]
60+
- libnvjpeg-dev {{ libnvjpeg_constraints }} # [win64]
61+
- cuda-cudart-dev {{ cuda_constraints }} # [win64]
62+
- cuda-nvrtc-dev {{ cuda_constraints }} # [win64]
63+
- libcufft-dev {{ libcufft_constraints }} # [win64]
64+
- libcublas-dev {{ libcublas_constraints }} # [win64]
65+
- libcusolver-dev {{ libcusolver_constraints }} # [win64]
66+
- libcusparse-dev {{ libcusparse_constraints }} # [win64]
67+
- libnpp-dev {{ libnpp_constraints }} # [win64]
5768
test:
5869
commands:
5970
- echo "pytorch-cuda metapackage is created."

0 commit comments

Comments
 (0)