mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-13 00:08:26 +08:00
Release v13.3 of the CUDA samples with CUDA 13.3 Toolkit (#435)
This is the release of the CUDA 13.3 samples, which include additions for CUDA Tile C++, and updated CCCL and Python samples.
This commit is contained in:
@@ -11,7 +11,7 @@ This sample demonstrates how to add a custom GPU operation to PyTorch using the
|
||||
- Python 3.10+
|
||||
- PyTorch 2.0+
|
||||
- cuda-python >= 13.0.0
|
||||
- cuda-core >= 0.6.0
|
||||
- cuda-core >=1.0.0
|
||||
|
||||
## Installation
|
||||
|
||||
@@ -20,6 +20,14 @@ cd python/3_FrameworkInterop/customPyTorchKernel
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
**Windows users:** The default `torch` wheel on PyPI for Windows is CPU-only and will cause `torch.cuda.is_available()` to return `False`. Install a CUDA-enabled build from PyTorch's wheel index *before* (or after) the command above:
|
||||
|
||||
```bash
|
||||
pip install torch --index-url https://download.pytorch.org/whl/cu128
|
||||
```
|
||||
|
||||
Replace `cu128` with the wheel suffix matching your installed CUDA driver (e.g. `cu121`, `cu124`, `cu126`, `cu128`). The driver's CUDA version must be >= the wheel's bundled runtime.
|
||||
|
||||
## How to Run
|
||||
|
||||
```bash
|
||||
|
||||
@@ -1,5 +1,10 @@
|
||||
# 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>=0.6.0
|
||||
cuda-core>=1.0.0
|
||||
torch>=2.0.0
|
||||
|
||||
@@ -22,7 +22,7 @@ Learn how to add a custom GPU operation to TensorFlow using `cuda.core` with `tf
|
||||
- Python 3.10+
|
||||
- TensorFlow 2.10+
|
||||
- cuda-python >= 13.0.0
|
||||
- cuda-core >= 0.6.0 (required for LEGACY_DEFAULT_STREAM)
|
||||
- cuda-core >=1.0.0 (required for LEGACY_DEFAULT_STREAM)
|
||||
- numpy >= 2.3.2
|
||||
- CuPy (for device pointer access)
|
||||
|
||||
|
||||
@@ -5,10 +5,10 @@
|
||||
# - TensorFlow 2.10+: Deep learning framework (tf.py_function, tf.custom_gradient)
|
||||
# - CuPy: Internal helper for device pointer access only
|
||||
#
|
||||
# Note: cuda-core>=0.6.0 is required for LEGACY_DEFAULT_STREAM constant
|
||||
# Note: cuda-core>=1.0.0 is required for LEGACY_DEFAULT_STREAM constant
|
||||
|
||||
numpy>=2.3.2
|
||||
tensorflow>=2.10.0
|
||||
cupy-cuda13x>=13.0.0
|
||||
cupy-cuda13x>=14.0.0
|
||||
cuda-python>=13.0.0
|
||||
cuda-core>=0.6.0
|
||||
cuda-core>=1.0.0
|
||||
|
||||
Reference in New Issue
Block a user