GitHubじゃ!Pythonじゃ!

GitHubからPython関係の優良リポジトリを探したかったのじゃー、でも英語は出来ないから日本語で読むのじゃー、英語社会世知辛いのじゃー

cupy

cupy – CUDA 加速されたNumPyのようなAPI

投稿日:

CUDA https://cupy.chainer.orgで 加速されたNumPyのようなAPI

CuPy:CUDAで加速されたNumPy様API

ウェブサイト | ドキュメント | インストールガイド | チュートリアル | 公式 )| フォーラムenja

CuPyは、CUDA上にNumPy互換の多次元配列を実装したものです。 CuPyは、コアの多次元配列クラス、 cupy.ndarray 、およびその上の多くの関数で構成されています。 numpy.ndarrayインターフェースのサブセットをサポートしています。

インストール

CuPyのインストールの詳細については、インストールガイドを参照してください

pipを使用してCuPyをインストールすることができます:

$ pip install cupy

CUDA、cuDNN、および/またはNCCLを有効にする場合は、CuPyをインストールする前に設定する必要があります。

ドッカーで走る

公式のDocker画像を提供します。 GPUでCuPyイメージを実行するには、 nvidia- dockerコマンドを使用します。 環境にbashでログインし、Pythonインタプリタを実行することができます。

$ nvidia-docker run -it cupy/cupy /bin/bash

開発

投稿ガイドをご覧ください。

詳しくは

ライセンス

MITライセンス( LICENSEファイルを参照)。

参照

奥方亮介、宇野雄也、西野大輔、堂本昌平、ルーミスクリスマン。 CuPy:NVIDIA GPU計算用NumPy互換ライブラリ 第30回神経情報処理システム学会年次総会(NIPS) (2017)の機械学習システム(LearningSys)に関するワークショップの議事録 URL

@inproceedings{cupy_learningsys2017,
  author       = "Okuta, Ryosuke and Unno, Yuya and Nishino, Daisuke and Hido, Shohei and Loomis, Crissman",
  title        = "CuPy: A NumPy-Compatible Library for NVIDIA GPU Calculations",
  booktitle    = "Proceedings of Workshop on Machine Learning Systems (LearningSys) in The Thirty-first Annual Conference on Neural Information Processing Systems (NIPS)",
  year         = "2017",
  url          = "http://learningsys.org/nips17/assets/papers/paper_16.pdf"
}







-cupy
-, , , , , , , , ,

執筆者: