Created
July 23, 2023 07:13
-
-
Save rsimd/3c2f61ce204fa47ef9b35a7d7720d357 to your computer and use it in GitHub Desktop.
`pyproject.toml` for `rye` to install `pytorch` that can use `cuda` (wip)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
[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" |
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
[email protected]