Skip to main content

Interface for using cupy in xarray, providing convenience accessors.

Project description

cupy-xarray

[!IMPORTANT] ⚠️ This project is looking for maintainers and contributors. Come help out!

GitHub Workflow CI Status pre-commit.ci status Documentation Status

PyPI Conda-forge

NASA-80NSSC22K0345

Interface for using cupy in xarray, providing convenience accessors.

Installation

From anaconda:

conda install cupy-xarray -c conda-forge

From PyPI:

pip install cupy-xarray

The latest version from Github:

pip install git+https://github.com/xarray-contrib/cupy-xarray.git

Usage

import xarray as xr
import cupy_xarray  # This registers the `DataSet.cupy` and `DataArray.cupy` namespaces but is not used directly

ds = xr.tutorial.load_dataset("air_temperature.nc")
type(ds.air.data)  # numpy.ndarray

%timeit ds.air.mean()  # 8.56 ms ± 15.6 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

ds = ds.cupy.as_cupy()  # Also available via convenience method ds.as_cupy()
type(ds.air.data)  # cupy.core.core.ndarray

%timeit ds.air.mean()  # 2.14 ms ± 21.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

ds = ds.as_numpy()
type(ds.air.data)  # numpy.ndarray

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cupy_xarray-0.1.4.tar.gz (25.6 kB view details)

Uploaded Source

Built Distribution

cupy_xarray-0.1.4-py3-none-any.whl (8.9 kB view details)

Uploaded Python 3

File details

Details for the file cupy_xarray-0.1.4.tar.gz.

File metadata

  • Download URL: cupy_xarray-0.1.4.tar.gz
  • Upload date:
  • Size: 25.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.19

File hashes

Hashes for cupy_xarray-0.1.4.tar.gz
Algorithm Hash digest
SHA256 3a0793efb90892bb56c7b5cc6d609732ee5d523169dd4925d76c372fffeababc
MD5 e4c46c63d5cb73c0e6b35dddf2177419
BLAKE2b-256 92f0217ee1161f069c79c9f4bda494fd151f5c23721e4948a1f7de284c8282b8

See more details on using hashes here.

File details

Details for the file cupy_xarray-0.1.4-py3-none-any.whl.

File metadata

  • Download URL: cupy_xarray-0.1.4-py3-none-any.whl
  • Upload date:
  • Size: 8.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.19

File hashes

Hashes for cupy_xarray-0.1.4-py3-none-any.whl
Algorithm Hash digest
SHA256 3023568519fabef9ded78857fae783e17d9574b1662ba53f9d953b84b00019bc
MD5 a37983f61640a0b22bfcb30d0f1603a1
BLAKE2b-256 fa250be9314cd72fe2ee2ef89ceb1f438bc156428a12177d684040456eee4a56

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page