tensorflow 在指定CPU/GPU上运行方法

转自:http://wiki.jikexueyuan.com/project/tensorflow-zh/how_tos/using_gpu.htmlhtml 使用 GPUs 支持的设备 在一套标准的系统上一般有多个计算设备. TensorFlow 支持 CPU 和 GPU 这两种设备. 咱们用指定字符串 strings 来标识这些设备. 好比:node "/cpu:0": 机器中的 CPU "/
相关文章
相关标签/搜索