search

openai / pixel-cnn

Code for the paper "PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications"
https://arxiv.org/abs/1701.05517
Other
1.92k stars 436 forks source link

Windows: No one_hot kernel in tensorflow-gpu-v1.0 #14

Open DawyD opened 7 years ago

DawyD commented 7 years ago

On windows, while running the code as it is provided I received the following error: Cannot assign a device to node 'one_hot': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.

There were some problems with one_hot kernel in tensorflow and believe that it is disabled in tensorflow version 1.0 ( https://github.com/tensorflow/tensorflow/pull/6822 )

Tensorflow: tensorflow-gpu 1.0.1, tensorflow-gpu 1.1.0rc0 OS: Windows 10, CUDA: 8.0, CUDNN: 5.1, Python 3.5, NVIDIA GTX 1080

Possible workaround is to allow soft-placement, but I am not sure how big impact on performance it has.

JunhyeopLee commented 7 years ago

I do have the same issue. Is it possible to have this issue when using only one GPU rather than multiple ones?