Skip to content

Instantly share code, notes, and snippets.

@rsimd
Created July 23, 2023 07:13
Show Gist options
  • Save rsimd/3c2f61ce204fa47ef9b35a7d7720d357 to your computer and use it in GitHub Desktop.
Save rsimd/3c2f61ce204fa47ef9b35a7d7720d357 to your computer and use it in GitHub Desktop.
`pyproject.toml` for `rye` to install `pytorch` that can use `cuda` (wip)
[project]
name = "dsbook"
version = "0.1.0"
description = "Add a short description here"
dependencies = [
"numpy>=1.24.4",
"scipy>=1.10.1",
"pandas>=2.0.3",
"matplotlib>=3.7.2",
"plotly>=5.15.0",
"tqdm>=4.65.0",
"seaborn>=0.12.2",
"ipykernel>=6.24.0",
"scikit-learn>=1.3.0",
"torch==2.0.1+cu118",
"opencv-python>=4.8.0.74",
"Pillow>=10.0.0",
]
readme = "README.md"
requires-python = ">= 3.11"
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.rye]
managed = true
[tool.hatch.metadata]
allow-direct-references = true
[[tool.rye.sources]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu118"
@rsimd
Copy link
Author

rsimd commented Jul 23, 2023

@Jiuh-star
Copy link

Currently rye use uv by default, but uv doesn't support --index-url and --extra-index-url due to potential security concern, so to make the above code work, setting behavior.use-uv=false in ~/.rye/config.toml is required.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment