diff --git a/conda/pytorch-cuda/conda_build_config.yaml b/conda/pytorch-cuda/conda_build_config.yaml new file mode 100644 index 000000000..802638b8e --- /dev/null +++ b/conda/pytorch-cuda/conda_build_config.yaml @@ -0,0 +1,3 @@ +version: + - 11.6 + - 11.7 diff --git a/conda/pytorch-cuda/meta.yaml b/conda/pytorch-cuda/meta.yaml new file mode 100644 index 000000000..92c970654 --- /dev/null +++ b/conda/pytorch-cuda/meta.yaml @@ -0,0 +1,43 @@ +{% set build = 0 %} +{% set cuda_constraints=">=11.6,<11.7" %} +{% if version == '11.7' %} +{% set cuda_constraints=">=11.7,<11.8" %} +{% endif %} + +package: + name: pytorch-cuda + version: {{ version }} +build: + number: {{ build }} + noarch: generic +requirements: + run: + - cuda={{ version }} + run_constrained: + - cuda-cccl {{ cuda_constraints }} + - cuda-command-line-tools {{ cuda_constraints }} + - cuda-compiler {{ cuda_constraints }} + - cuda-cudart {{ cuda_constraints }} + - cuda-cudart-dev {{ cuda_constraints }} + - cuda-cuobjdump {{ cuda_constraints }} + - cuda-cupti {{ cuda_constraints }} + - cuda-cuxxfilt {{ cuda_constraints }} + - cuda-driver-dev {{ cuda_constraints }} + - cuda-libraries {{ cuda_constraints }} + - cuda-libraries-dev {{ cuda_constraints }} + - cuda-cudaart-dev {{ cuda_constraints }} + - cuda-nvcc {{ cuda_constraints }} + - cuda-nvml-dev {{ cuda_constraints }} + - cuda-nvprune {{ cuda_constraints }} + - cuda-nvrtc {{ cuda_constraints }} + - cuda-nvrtc-dev {{ cuda_constraints }} + - cuda-nvtx {{ cuda_constraints }} + - cuda-runtime {{ cuda_constraints }} + - cuda-toolkit {{ cuda_constraints }} + - cuda-tools {{ cuda_constraints }} + # None, pytorch should depend on pytorch-cuda +test: + commands: + - echo "pytorch-cuda metapackage is created." +about: + summary: Metapackage to select the PyTorch variant. Use conda's pinning mechanism in your environment to control which variant you want.