diff --git a/packaging/build_conda.py b/packaging/build_conda.py index c2c7f1b2..b3b3bdaa 100644 --- a/packaging/build_conda.py +++ b/packaging/build_conda.py @@ -123,7 +123,7 @@ def do_build(start_args: List[str]) -> None: if test_flag is not None: args.append(test_flag) - args.extend(["-c", "bottler", "-c", "iopath", "-c", "conda-forge"]) + args.extend(["-c", "bottler", "-c", "iopath", "-c", "conda-forge", "-c", "nvidia/label/cuda-12.1.0"]) args.append("--no-anaconda-upload") args.extend(["--python", os.environ["PYTHON_VERSION"]]) args.append("packaging/pytorch3d") diff --git a/packaging/pytorch3d/meta.yaml b/packaging/pytorch3d/meta.yaml index 3a0bfbc5..80e4b0c7 100644 --- a/packaging/pytorch3d/meta.yaml +++ b/packaging/pytorch3d/meta.yaml @@ -9,6 +9,7 @@ requirements: build: - {{ compiler('c') }} # [win] - cuda-toolkit + - cuda-version=12.1 {{ environ.get('CONDA_CUB_CONSTRAINT') }} host: