You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
Commit f1185c7
Browse filesBrowse the repository at this point in the historyBrowse files
The CUDA backend page and the getting started page said several things
that the TensorRT page, and the code, contradict:
- The CUDA page said the Python runtime allocates host memory for an
unplanned output, so a method that keeps its outputs on the GPU could not
run from Python. The Python runtime allocates a new device buffer for each
call.
- It said a planned input fed with GPU memory makes an undefined host
memcpy and crashes. The runtime refuses that copy and the call fails with
NotSupported.
- It asked for a CUDA 11.x or later toolkit to run a model. The prebuilt
packages use CUDA 13 and bring the CUDA runtime with them on Linux, a
build from source also supports CUDA 12.6, and the toolkit is needed for
export, where nvcc compiles the kernels. It also gave no minimum GPU,
while the packages are built for compute capability 8.0 and newer, and it
asked for Python 3.8+, while the package needs 3.11 to 3.15.
- It listed Windows without saying that programs for Windows are exported
on Linux. It now links to the C++ guide section that shows how. Its link
to the getting started page was also broken.
- The getting started page said to use the CPU packages on Windows, while
CUDA packages are built for Windows too, and it called CUDA 13.4 nightly
only.
To develop and export models using the CUDA backend:
25
27
26
-
-**Python**: Python 3.8+
28
+
-**Python**: Python 3.11 to 3.15
27
29
-**PyTorch**: PyTorch with CUDA support
30
+
-**CUDA Toolkit**: export compiles kernels with `nvcc`, the CUDA compiler, so it needs a CUDA toolkit
28
31
-**ExecuTorch**: Install ExecuTorch with CUDA backend support
29
32
30
33
## Using the CUDA Backend
@@ -83,7 +86,7 @@ For a complete working example, see the [CUDA export script](https://github.com/
83
86
84
87
## Runtime Integration
85
88
86
-
To run the model on device, use the standard ExecuTorch runtime APIs. See [Running on Device](getting-started.md#running-on-device) for more information.
89
+
To run the model on device, use the standard ExecuTorch runtime APIs. See [Running on Device](../../getting-started.md#running-on-device) for more information.
87
90
88
91
When building from source, pass `-DEXECUTORCH_BUILD_CUDA=ON` when configuring the CMake build to compile the CUDA backend.
0 commit comments