TensorFlow之设备(device)详解

对TensorFlow不了解的,猛的一看还以为TensorFlow是个设备什么的,其实本文要说的是Python的TensorFlow模块下一个叫device()的方法,即: import tensorflow as tf with tf.device('/gpu:0'): ... ... ... 设备是指一块可以用来运算并且拥有自己的地址空间的硬件,如CPU和GPU Te
相关文章
相关标签/搜索