Skip to content

Instantly share code, notes, and snippets.

@Kjdragan

Kjdragan/install_pytorch_w_uv.toml

Created January 3, 2025 22:58
Show Gist options
  • Save Kjdragan/6218477f5ea9399528bb20dfe2bef9a8 to your computer and use it in GitHub Desktop.
Save Kjdragan/6218477f5ea9399528bb20dfe2bef9a8 to your computer and use it in GitHub Desktop.
Download ZIP
Pytorch
Raw
install_pytorch_w_uv.toml
[project]
name = "smolagents-playground"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"torch>=2.5.1",
"torchvision>=0.20.1",
]
[tool.uv.sources]
torch = [
{ index = "pytorch-cu124", marker = "platform_system != 'Darwin'"},
]
torchvision = [
{ index = "pytorch-cu124", marker = "platform_system != 'Darwin'"},
]
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment