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
+ #
2
8
# Please note: Build number should be advanced with
3
9
# every deployment. After the deployment to production
4
10
# use following links to validate the correctness of
@@ -36,15 +42,11 @@ package:
36
42
version : {{ version }}
37
43
build :
38
44
number : {{ build }}
39
- noarch : generic
40
45
requirements :
41
46
run :
42
47
- cuda-libraries {{ cuda_constraints }}
43
48
- cuda-nvtx {{ cuda_constraints }}
44
49
- 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 :
48
50
- cuda-cudart {{ cuda_constraints }}
49
51
- cuda-cupti {{ cuda_constraints }}
50
52
- cuda-nvrtc {{ cuda_constraints }}
@@ -54,6 +56,15 @@ requirements:
54
56
- libcusolver {{ libcusolver_constraints }}
55
57
- libcusparse {{ libcusparse_constraints }}
56
58
- 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]
57
68
test :
58
69
commands :
59
70
- echo "pytorch-cuda metapackage is created."
0 commit comments