NASP/cuda-samples
85
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2026-10-26 06:38:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Sync v13.3-public from internal repo

Browse Source
This commit is contained in:
cuda-samples sync
2026-05-27 21:03:57 +00:00
parent aeab82ff30
commit 99686e0c71
150 changed files with 7977 additions and 246 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/3_FrameworkInterop/customTensorFlowKernel/README.md
View File

@@ -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)

6
python/3_FrameworkInterop/customTensorFlowKernel/requirements.txt
View File

@@ -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