1
0
mirror of https://github.com/tiyn/wiki.git synced 2026-09-13 10:31:36 +02:00

Python: Added CuPy

This commit is contained in:
2026-09-06 12:00:44 +02:00
parent ef929292da
commit dae29c2a08

View File

@@ -275,6 +275,23 @@ cuml.accel.install()
Afterward all possible scikit-learn algorithms will run on the GPU instead of the CPU.
### NumPy
[NumPy](https://numpy.org/) is a free and open-source library for numerical computing with Python.
By default, NumPy only utilizes the CPU.
#### Run NumPy Operations on the GPU
Using [CuPy](https://cupy.dev/), NumPy-like operations can be executed on the GPU.
CuPy provides an API largely compatible with NumPy and can therefore often be used by simply
replacing the NumPy import.
```py
import cupy as np
```
CuPy supports Nvidia GPUs via CUDA and AMD GPUs via ROCm.
### PyTorch
This section addresses the [PyTorch module](https://pytorch.org/).