# Custom PyTorch Kernel Sample Requirements
#
# NOTE: On Windows, the default `torch` wheel from PyPI is CPU-only and the
# sample will fail with "Torch not compiled with CUDA enabled". Install a
# CUDA-enabled torch from PyTorch's wheel index first (see README.md):
#   pip install torch --index-url https://download.pytorch.org/whl/cu128

cuda-python>=13.0.0
cuda-core>=1.0.0
torch>=2.0.0