我正在尝试将pylearn2
GPU模型转换为CPU兼容版本,以便在远程服务器上进行预测 - 如何将CudaNdarraySharedVariable
转换为TensorVariable
的避免在无GPU的机器上调用cuda代码时出错?实验性的theano标志unpickle_gpu_to_cpu
似乎留下了一些CudaNdarraySharedVariable
,特别是model.layers[n].transformer._W
)。
答案 0 :(得分:1)
对于一个普通的CudaNdarray变量,这样的东西应该可以工作:
'''x = CudaNdarray ... x_new = theano.tensor.TensorVariable(CudaNdarrayType([False] * tensor_dim))
f = theano.function([x_new],x_new)
converted_x = f(x) '''