Tensorflow中GPU和CPU的切换配置

最新推荐文章于 2024-02-02 15:46:59 发布

江户川柯西い

最新推荐文章于 2024-02-02 15:46:59 发布

阅读量1.4k

点赞数 3

分类专栏： Tensorflow 文章标签： Tensorflow 机器学习

本文链接：https://blog.csdn.net/zjy_cnds/article/details/103410739

版权

Tensorflow 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

使用CPU

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu，不注释就是使用gpu

一般情况安装过GPU的会默认使用gpu

使用GPU

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# os.environ['CUDA_VISIBLE_DEVICES'] = "0" #这个是仅选择使用GPU 0
# os.environ['CUDA_VISIBLE_DEVICES'] = "1" #这个是仅选择使用GPU 1
# os.environ['CUDA_VISIBLE_DEVICES']  = “0,1” #设置当前使用的GPU设备为0,1号