Skip to content

Commit 942c854

Browse files
committed
Slim down pytorch-cuda
It should only contain runtime dependencies that PyTorch+domain libraries depend on, namely: - cudart - cublas - cusparse - cufft - curand - nvtx - nvrtc - nvjpeg (for TorchVision) This removes dependencies on NVCC, build/debug tools, etc which are not needed for running the pytorch Test Plan: `conda create -n tmp -c nvidia -c malfet cuda-toolkit==11.7` and observe that only relevant packages are installed Fixes pytorch/pytorch#91334
1 parent c85d235 commit 942c854

File tree

1 file changed

+2
-14
lines changed

1 file changed

+2
-14
lines changed

conda/pytorch-cuda/meta.yaml

+2-14
Original file line numberDiff line numberDiff line change
@@ -39,31 +39,19 @@ build:
3939
noarch: generic
4040
requirements:
4141
run:
42-
- cuda={{ version }}
42+
- cuda-libraries {{ cuda_constraints }}
43+
- cuda-nvtx {{ cuda_constraints }}
4344
run_constrained:
44-
- cuda-cccl {{ cuda_constraints }}
45-
- cuda-command-line-tools {{ cuda_constraints }}
46-
- cuda-compiler {{ cuda_constraints }}
4745
- cuda-cudart {{ cuda_constraints }}
48-
- cuda-cudart-dev {{ cuda_constraints }}
49-
- cuda-cuobjdump {{ cuda_constraints }}
5046
- cuda-cupti {{ cuda_constraints }}
51-
- cuda-cuxxfilt {{ cuda_constraints }}
52-
- cuda-libraries {{ cuda_constraints }}
53-
- cuda-nvcc {{ cuda_constraints }}
54-
- cuda-nvprune {{ cuda_constraints }}
5547
- cuda-nvrtc {{ cuda_constraints }}
56-
- cuda-nvtx {{ cuda_constraints }}
5748
- cuda-runtime {{ cuda_constraints }}
58-
- cuda-toolkit {{ cuda_constraints }}
59-
- cuda-tools {{ cuda_constraints }}
6049
- libcufft {{ libcufft_constraints }}
6150
- libcublas {{ libcublas_constraints }}
6251
- libcusolver {{ libcusolver_constraints }}
6352
- libcusparse {{ libcusparse_constraints }}
6453
- libnpp {{ libnpp_constraints }}
6554
- libnvjpeg {{ libnvjpeg_constraints }}
66-
# None, pytorch should depend on pytorch-cuda
6755
test:
6856
commands:
6957
- echo "pytorch-cuda metapackage is created."

0 commit comments

Comments
 (0)