Add cuda-nvtx-dev

facebookincubator · Jan 18, 2025 · 48491ea · 48491ea
1 parent 9cd6a01
commit 48491ea
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 5 deletions.
diff --git a/pixi.lock b/pixi.lock
diff --git a/pixi.toml b/pixi.toml
@@ -219,6 +219,7 @@ build_py = { cmd = "pip install . -vv", env = { CMAKE_ARGS = """
 #=========
 
 [target.win-64.dependencies]
+cuda-nvtx-dev = ">=12.6.77,<13"
 
 [target.win-64.tasks]
 config = { cmd = """
@@ -297,19 +298,19 @@ platforms = ["linux-64", "osx-64", "osx-arm64", "win-64"]
 channels = ["conda-forge"]
 platforms = ["linux-64"]
 system-requirements = { cuda = "12.0" }
-dependencies = { cuda-toolkit = "12.*", python = "3.12.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } }
+dependencies = { cuda-toolkit = "12.*", python = "3.12.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } , cuda-nvtx-dev = ">=12.6.77,<13" }
 
 [feature.py311-cuda126]
 channels = ["conda-forge"]
 platforms = ["linux-64"]
 system-requirements = { cuda = "12.0" }
-dependencies = { cuda-toolkit = "12.*", python = "3.11.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } }
+dependencies = { cuda-toolkit = "12.*", python = "3.11.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } , cuda-nvtx-dev = ">=12.6.77,<13" }
 
 [feature.py310-cuda126]
 channels = ["conda-forge"]
 platforms = ["linux-64"]
 system-requirements = { cuda = "12.0" }
-dependencies = { cuda-toolkit = "12.*", python = "3.10.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } }
+dependencies = { cuda-toolkit = "12.*", python = "3.10.*", pytorch-gpu = { version = ">=2.5.1,<3", build = "cuda126_*" } , cuda-nvtx-dev = ">=12.6.77,<13" }
 
 #==============
 # Environments