cupy-cuda12x

CuPy: NumPy & SciPy for GPU

19 个版本 Python >=3.10
安装
pip install cupy-cuda12x
poetry add cupy-cuda12x
pipenv install cupy-cuda12x
conda install cupy-cuda12x
描述

.. image:: https://raw.githubusercontent.com/cupy/cupy/main/docs/image/cupy_logo_1000px.png :width: 400

CuPy : NumPy & SciPy for GPU

CuPy <https://cupy.dev/>_ is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python.

This is a CuPy wheel (precompiled binary) package for CUDA 12.x. You need to install CUDA Toolkit 12.x <https://developer.nvidia.com/cuda-toolkit-archive>_ locally to use these packages. Alternatively, you can install this package together with all needed CUDA components from PyPI by passing the [ctk] tag::

$ pip install cupy-cuda12x[ctk]

If you have another version of CUDA, or want to build from source, refer to the Installation Guide <https://docs.cupy.dev/en/latest/install.html>_ for instructions.