1. 首页
  2. 编程语言
  3. Python
  4. 详解tf.device()指定tensorflow运行的GPU或CPU设备实现

详解tf.device()指定tensorflow运行的GPU或CPU设备实现

上传者: 2021-06-11 18:47:24上传 PDF文件 244.48 KB 热度 13次

在tensorflow中,我们可以使用 设置使用GPU使用 tf.device 指定Session在第二块GPU上运行:ConfigProto() 中参数 如果安装的是GPU版本的tensorflow,机器上有支持的GPU,也正确安装了显卡驱动、CUDA和cuDNN,默认情况下,Session会在GPU上运行:默认在GPU:0上执行:设置使用cputensorflow中不同的GPU使用/gpu:0和/gpu:1区分,而CPU不区分设备号,统一使用 /cpu:0

下载地址
用户评论