Skip to content

Commit

Permalink
Force jaxlib CUDA build with pixi GPU environment
Browse files Browse the repository at this point in the history
Co-authored-by: Diego Ferigo <diego.ferigo@iit.it>
  • Loading branch information
flferretti and diegoferigo committed Aug 21, 2024
1 parent 9ff3094 commit aa532e2
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -206,10 +206,14 @@ pytest-icdiff = "*"
robot_descriptions = "*"

[tool.pixi.feature.gpu]
dependencies = { cuda-version = "12.*", cuda-cupti = "*" }
platforms = ["linux-64"]
system-requirements = { cuda = "12.1" }

[tool.pixi.feature.gpu.dependencies]
cuda-cupti = "*"
cuda-version = "12.*"
jaxlib = {version = "*", build = "*cuda*"}

[tool.pixi.pypi-dependencies]
jaxsim = { path = "./", editable = true }

Expand Down

0 comments on commit aa532e2

Please sign in to comment.