我正在尝试运行github上可用的pixel-cnn神经网络。按照README.md中的说明，我在cmd中运行以下代码：
train.py -i ./data_dir/ -o ./save_dir -g 1
我正在使用一个GPU，并在与train.py相同的目录中创建了两个文件夹./data_dir和./save_dir，用于加载和保存数据。这样做时，我收到以下错误消息：
tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation model_1/ones: node model_1/ones (defined at \OneDrive - MNG\Matura Arbeit\Projects\pixel-cnn-master\pixel_cnn_pp\model.py:36) was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device. The requested device appears to be a GPU, but CUDA is not enabled.
似乎tensorflow无法识别GPU，但是在检查可用于tensorflow的设备时（如here所述），我的CPU和GPU都显示为“ / device：CPU：0”和/ device ：GPU：0“。此外，当使用tensorflow-gpu运行其他程序时，它可以正常运行。
我已经安装了 tensorflow-gpu == 1.14.0 。至于CUDA，我敢肯定我已经安装了10.0版，如nvcc --version所示。尽管在运行nvidia-smi时它表明已安装 CUDA版本10.1 。
（edited :)我正在将Anaconda环境（Windows 10）与 tensorflow-gpu == 1.14.0 一起使用。我使用的GPU是具有Max-Q设计和驱动程序版本436.30 的 GTX 1050Ti 。至于CUDA，我敢肯定我已经安装了版本10.0 ，如nvcc --version所示。尽管在运行nvidia-smi时它表明已安装 CUDA版本10.1 。