tiyn/wiki
1
0
Fork 0
mirror of https://github.com/tiyn/wiki.git synced 2026-09-13 10:31:36 +02:00
Code Issues Projects Releases Wiki Activity

Python: Added CuPy

Browse Source
This commit is contained in:
Marten
2026-09-06 12:00:44 +02:00
parent ef929292da
commit dae29c2a08
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
wiki/programming_language/python.md
View File

@@ -275,6 +275,23 @@ cuml.accel.install()
Afterward all possible scikit-learn algorithms will run on the GPU instead of the CPU. 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 ### PyTorch
This section addresses the [PyTorch module](https://pytorch.org/). This section addresses the [PyTorch module](https://pytorch.org/).