PyTorch, установленный через проект uv, показывает версию только для ЦП в Windows со спецификацией CUDA в pyproject.toml ⇐ Python

[Anonymous]

Я пытаюсь настроить проект Python с использованием uv и pyproject.toml в Windows. Я хочу установить PyTorch с поддержкой CUDA, но после установки, когда я проверяю версию, он показывает только процессор.
Вот моя настройка pyproject.toml:

Код: Выделить всё

[project]
name = "transformerpractice"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
"en-core-web-sm",
"ko-core-news-sm",
"openpyxl>=3.1.5",
"pandas>=2.3.3",
"spacy>=3.8.8",
"torchtext==0.16.0",
"torch==2.1.0",
]

[tool.uv.sources]
en-core-web-sm = { url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0.tar.gz" }
ko-core-news-sm = { path = "tokenizer/ko_core_news_sm-3.8.0-py3-none-any.whl" }
torch = [
{ index = "pytorch-cu121", marker = "sys_platform == 'win_amd64'" },
]

[[tool.uv.index]]
name = "pytorch-cu121"
url = "https://download.pytorch.org/whl/cu121"
explicit = false

После установки запускаю:

Код: Выделить всё

import torch

print(torch.__version__)
print(torch.version.cuda)

И вывод:

Код: Выделить всё

2.1.0+cpu
None

Похоже, что установлена версия CPU, а не версия CUDA.
Что я пробовал:

• Указание версии PyTorch с помощью torch==2.1.0 в [project.dependents].
• Добавление пользовательского PyTorch индексировать с поддержкой CUDA с помощью [tool.uv.index].
• Проверяю, что я использую 64-разрядную версию Windows (

  Код: Выделить всё

  sys_platform == 'win_amd64'

  ).

Вопрос:
Как настроить pyproject.toml с помощью uv, чтобы он устанавливал PyTorch (2.1.0+cu121) с поддержкой CUDA в Windows?
Любые указания о том, как uv обрабатывает дополнительные индексы и маркеры платформы для PyTorch были бы полезны.

Подробнее здесь: https://stackoverflow.com/questions/798 ... h-cuda-spe