我有两台Windows计算机,一台有GPU,一台没有GPU。
我想在两者上部署相同的python脚本(TensorFlow 1.8对象检测),而无需更改TensorFlow的打包版本。换句话说,我想在CPU上运行tensorflow-gpu。
在我的脚本无法检测到nvcuda.dll
的情况下,我尝试使用会话配置禁用GPU,如下所示:
config = tf.ConfigProto(
device_count = {'GPU': 0}
)
和:
with tf.Session(graph=detection_graph, config=config) as sess:
但是,这还不够,因为TensorFlow仍然返回错误:
ImportError: Could not find 'nvcuda.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable.
Typically it is installed in 'C:\Windows\System32'. If it is not present, ensure that you have a CUDA-capable GPU with the correct driver installed.
有什么方法可以完全禁用对GPU / CUDA的检查并默认为CPU?
编辑:我已经阅读了有关Linux posted here上tensorflow-gpu == 1.0的答案,这表明这是不可能的。我很想知道这是否仍然是tensorflow-gpu的编译方式(9个版本以后)。