将CudaNdarraySharedVariable转换为TensorVariable

时间:2014-07-14 19:49:54

标签: python machine-learning neural-network gpu theano

我正在尝试将pylearn2 GPU模型转换为CPU兼容版本,以便在远程服务器上进行预测 - 如何将CudaNdarraySharedVariable转换为TensorVariable的避免在无GPU的机器上调用cuda代码时出错?实验性的theano标志unpickle_gpu_to_cpu似乎留下了一些CudaNdarraySharedVariable,特别是model.layers[n].transformer._W)。

1 个答案:

答案 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) '''