使用Theano配置cnmem =1导入theano失败

Question

使用Theano配置cnmem =1导入theano失败

知道如何确保GPU完全分配给theano python脚本吗？

注意：显示不用于避免其GPU的使用

文件：.theanorc

cnmem = 1

文件: test.py

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

输出：

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

它只适用于cnmem = 0.75。

文件：.theanorc

cnmem = 0.75

输出：

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

Answer 1

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

你能试试1.0而不是1吗？根据文档，它需要一个浮子。此外，它被限制在0.95，以便允许设备驱动程序的空间。所以，你不能使用整个GPU内存，就像你不能使用所有的RAM一样。