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!
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)
Built Distribution
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a0793efb90892bb56c7b5cc6d609732ee5d523169dd4925d76c372fffeababc |
|
MD5 | e4c46c63d5cb73c0e6b35dddf2177419 |
|
BLAKE2b-256 | 92f0217ee1161f069c79c9f4bda494fd151f5c23721e4948a1f7de284c8282b8 |
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3023568519fabef9ded78857fae783e17d9574b1662ba53f9d953b84b00019bc |
|
MD5 | a37983f61640a0b22bfcb30d0f1603a1 |
|
BLAKE2b-256 | fa250be9314cd72fe2ee2ef89ceb1f438bc156428a12177d684040456eee4a56 |