已将操作明确分配给/ GPU:1,但可用设备为

时间:2018-09-25 21:46:20

标签: tensorflow select gpu

我的系统有2个Nvidia GPU,一个分配给OS的OS图形用GTX 750 Ti和一个免费用于Tensorflow的GTX 1080Ti。我使用呼叫:tensorflow::graph::SetDefaultDevice("/GPU:1", &graph);启用GPU1。

我正在运行为C ++ / CMake和Cuda编译工具(版本9.1,V9.1.85)手动编译和配置的TF 1.10。当我分配GPU 1执行图形时,出现以下错误:

  

”无效的参数:无法为操作“ h1_w / read”分配设备:   已将操作明确分配给/ GPU:1,但可用设备为   [/ job:localhost / replica:0 / task:0 / device:CPU:0,   / job:localhost /副本:0 / task:0 / device:GPU:0]。确保设备   规范是指有效的设备。 [[{{node h1_w / read}} =   IdentityT = DT_FLOAT,_class = [“ loc:@ h1_w”],_ device =“ / GPU:1”]]“

0 个答案:

没有答案