您当前的位置：首页 > IT编程 > python
\| C语言 \| Java \| VB \| VC \| python \| Android \| TensorFlow \| C++ \| oracle \| 学术与代码 \| cnn卷积神经网络 \| gnn \| 图像修复 \| Keras \| 数据集 \| Neo4j \| 自然语言处理 \| 深度学习 \| 医学CAD \| 医学影像 \| 超参数 \| pointnet \| pytorch \| 异常检测 \|

自学教程：Tensorflow训练模型默认占满所有GPU的解决方案

51自学网 2023-06-16 18:56:01

python

这篇教程Tensorflow训练模型默认占满所有GPU的解决方案写得很实用，希望能帮到您。

Tensorflow训练模型默认占满所有GPU问题

在使用gpu服务器训练tensorflow模型时，总是占满显存！

TensorFlow默认的是占用所有GPU

因此我们需要手动设置使用的GPU编号以及单个GPU显存占用比例

1.第一步需要在代码中开头加入

import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 按照PCI_BUS_ID顺序从0开始排列GPU设备os.environ["CUDA_VISIBLE_DEVICES"]=‘0' 
pytorch版本PSEnet训练并部署方式
Python+SQLAlchemy轻松实现管理数据库

自学教程：Tensorflow训练模型默认占满所有GPU的解决方案

目录

Tensorflow训练模型默认占满所有GPU问题

TensorFlow默认的是占用所有GPU