force cuda versions

facebookresearch · Oct 29, 2024 · 56885b4 · 56885b4
1 parent 9bdf0b3
commit 56885b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -9,6 +9,7 @@ requirements:
   build:
     - {{ compiler('c') }} # [win]
     - cuda-toolkit
+    - cuda-version=12.1
     {{ environ.get('CONDA_CUB_CONSTRAINT') }}
 
   host: