如何在Visual Studio 2017中的Python / TensorFlow项目中启用GPU支持?

时间:2017-08-17 12:25:49

标签: python visual-studio tensorflow gpu

我在Visual Studio 2017中安装了Python(Anaconda 4.4.0),并使用TensorFlow包的CPU版本从https://github.com/anishathalye/neural-style运行Neural-Style示例。

我正在尝试在安装TensorFlow-GPU软件包(安装了CUDA和CuDNN)后重新运行该示例,但我似乎无法让Visual Studio / Python认可tensorflow-gpu软件包。

如果我只是安装gpu软件包,解释器会暂停import tensorflow as tf行的执行,并报告无法找到导入。

是否有运行时标志,或对Python使用启用CUDA的TensorFlow版本所需的代码进行了一些修改?

Visual Studio 2017, Python project using TensorFlow

0 个答案:

没有答案