Python 像素cnn（tensorflow gpu）无法识别gpu

我正在尝试运行上可用的像素cnn神经网络。按照README.md中的说明，我在cmd中运行以下代码：

train.py -i ./data_dir/ -o ./save_dir -g 1

我使用一个gpu，在train.py的同一目录中创建了两个文件夹./data\u dir和./save\u 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可用的设备时（如上所述），我的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版

---

（编辑：）我使用的是蟒蛇环境（Windows 10）使用tensorflow gpu==1.14.0。我使用的gpu是GTX 1050Ti，采用Max-Q设计，驱动程序版本为436.30。至于CUDA，我非常确定我已经安装了10.0版，如

nvcc--version

所示。尽管运行

nvidia smi

时，它显示CUDA版本10.1已经安装。

只需检查您的CUDA计算级别即可卡片？你能用gpu版本运行tensorflow命令吗？@JCooke我使用的是GTX 1050Ti（带有Max-Q设计）。我现在看到它没有列为兼容，这对我来说很奇怪。这真的是这样吗，1050Ti不能用CUDA计算？（编辑：）是的，我运行的任何其他程序都可以与tensorflow-gpu完美配合。不，我认为列表中没有。应该支持。您使用的是虚拟环境/conda吗？您在路径中指向的是什么版本的CUDA？请您也用有关您的开发环境的更多信息更新您的问题，因为此错误可能是由CUDA引起的链接问题，尽管奇怪，但它只发生在这个文件上。@JCooke是的，我正在使用一个anaconda环境（我相应地更新了问题）.PATH设置为指向…\CUDA\v10.0\bin，CUDA\u设置为指向…\CUDA\v10.0。此外，为了检查tf gpu是否真的与其他文件一起工作，我快速运行，在gpu承载所有负载的情况下执行得非常好（通过任务管理器检查）