[ トップページ ]
数学・計算・情報学・プログラミング

GPU メモリに 1 バイト単位でつめこんだデータで機械学習できるようにした

Theano で GPU メモリに int8 配列をつめこむ方法」 というページで,(NVIDIA) GPU メモリにデータを 1 バイト単位でつめこんで,それをつかうたびに 32 bit 浮動小数に変換する方法を書いた. 余分な計算をしなければならないが,オーバヘッドはそれほどでもない. この方法をつかえば,かぎられた GPU メモリを有効につかうことができる.

この方法は Theano というシステムを前提にしている. 他のシステム (たとえば Caffe) にはべつの方法が必要だ. プログラムを CUDA で書けばできることはまちがいないが,高級言語で書いたほうがあつかいやすいこともまちがいない.

キーワード:

トラックバック

このエントリーのトラックバックURL:
https://www.kanadas.com/mt/mt-tb.cgi/7087

コメントを投稿

Google でブログを検索:

メインページアーカイブページも見てください.
Creative Commons License
このブログはつぎのライセンスで保護されています. クリエイティブ・コモンズ・ライセンス.
Powered by Movable Type