如何在AN中使用GPU时找到哪个变量是float64

2024-10-01 05:03:35 发布

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

在no中,使用gpu时,变量必须是float32。我检查了我所有的变量都是folat32,但是我仍然得到下面的错误。在

ERROR (theano.gof.opt): Optimization failure due to: local_gpu_incsubtensor
ERROR (theano.gof.opt): TRACEBACK:
ERROR (theano.gof.opt): Traceback (most recent call last):
File "/public/home/ybfan/anaconda/lib/python2.7/site-packages/theano/gof/opt.py", line 1286, in process_node
    replacements = lopt.transform(node)
File "/public/home/ybfan/anaconda/lib/python2.7/site-packages/theano/sandbox/cuda/opt.py", line 855, in local_gpu_incsubtensor
    gpu_from_host(y),
File "/public/home/ybfan/anaconda/lib/python2.7/site-packages/theano/gof/op.py", line 399, in __call__
    node = self.make_node(*inputs, **kwargs)
File "/public/home/ybfan/anaconda/lib/python2.7/site-packages/theano/sandbox/cuda/basic_ops.py", line 128, in make_node
    dtype=x.dtype)()])
File "/public/home/ybfan/anaconda/lib/python2.7/site-packages/theano/sandbox/cuda/type.py", line 68, in __init__
    (self.__class__.__name__, dtype, name))
TypeError: CudaNdarrayType only supports dtype float32 for now. Tried using dtype float64 for variable None

似乎有些变量仍然是float64,我的问题是如何定位我使用float64变量的位置。在


Tags: inpynodehomelibpackageslinesite