python - Theano 导入失败，theano 配置cnmem = 1

Question

Theano 导入失败，theano 配置cnmem = 1

知道如何确保将 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
$

score 1 · Accepted Answer

您可以尝试使用 1.0 而不是 1 吗？根据文档，它需要是一个浮点数。此外，它被限制为 0.95，以便为设备驱动程序留出空间。因此，您不能使用整个 GPU 内存，就像您不能使用所有 RAM 一样。

1 回答 1