2

theano 構成 cnmem = 1 で Theano のインポートに失敗する

GPUがtheano pythonスクリプトに完全に割り当てられていることを確認する方法はありますか?

注: GPU の使用を避けるためにディスプレイは使用されません。

ファイル: .theanrc

cnmem = 1

ファイル: test.py

print 'Importing Theano Library ...'
import theano
print 'Imported'

出力：

$ python test.py 
Importing Theano Library ... 
Killed 
$

cnmem = 0.75 でのみ動作します

ファイル: .theanrc

cnmem = 0.75

出力：

$ python test.py
Importing Theano Library ...
Imported
$
4

1 に答える 1

1

https://github.com/Theano/Theano/issues/4302#issuecomment-202067917

1 の代わりに 1.0 を試していただけますか? ドキュメントによると、フロートである必要があります。また、デバイス ドライバ用のスペースを確保するために、0.95 に制限されています。したがって、すべての RAM を使用できないのと同じように、GPU メモリ全体を使用することはできません。

于 2016-09-20T19:15:12.437 に答える