2014-07-14 36 views
1

我想一個pylearn2 GPU模型轉換爲CPU兼容的版本,預測在遠程服務器上 - 我怎麼能轉換CudaNdarraySharedVariable's到TensorVariable的,以避免錯誤在調用CUDA代碼無GPU的機器?實驗theano旗unpickle_gpu_to_cpu似乎已經留下了幾個CudaNdarraySharedVariable的掛起(特別是model.layers[n].transformer._W)。轉換CudaNdarraySharedVariable到TensorVariable

回答

1

對於一個普通的可變CudaNdarray,像這樣應該工作:

'''X = CudaNdarray ... x_new = theano.tensor.TensorVariable(CudaNdarrayType([FALSE] * tensor_dim))
F = theano.function([x_new],x_new)

converted_x = f(x) '''