mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-05-14 14:06:53 +08:00
aeab82ff30
- Added Python samples for CUDA Python 1.0 release - Renamed top-level `Samples` directory to `cpp` to accommodate Python samples.
15 lines
490 B
Plaintext
15 lines
490 B
Plaintext
# TensorFlow Custom GPU Operator using cuda.core
|
|
#
|
|
# This sample demonstrates a cuda.core-first approach:
|
|
# - cuda-python and cuda-core: Primary GPU management (device, stream, kernel)
|
|
# - 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
|
|
|
|
numpy>=2.3.2
|
|
tensorflow>=2.10.0
|
|
cupy-cuda13x>=13.0.0
|
|
cuda-python>=13.0.0
|
|
cuda-core>=0.6.0
|