无法导入无配置cnmem=1的无

2024-09-29 17:20:58 发布

您现在位置:Python中文网/ 问答频道 /正文

无法导入无配置cnmem=1的无

你知道如何确保GPU完全分配给TheanoPython脚本吗?你知道吗

Note: Display is not used to avoid its GPU usage

文件:.theanoc

cnmem = 1

文件:测试.py

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

输出:

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

它只适用于cnmem=0.75

文件:.theanoc

cnmem = 0.75

输出:

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

Tags: 文件pytest脚本gpudisplaylibrarytheano
1条回答
网友
1楼 · 发布于 2024-09-29 17:20:58

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

Could you try with 1.0 instead of 1? according to the docs, it needs to be a float. Also, it is limited to 0.95 to allow space for device drivers. So, you can't use the entire GPU memory just like you can't use all of RAM.

相关问题 更多 >

    热门问题