mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-06-04 16:26:53 +08:00
11 lines
391 B
Plaintext
11 lines
391 B
Plaintext
# 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
|