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

自学教程：tensorflow之如何使用GPU而不是CPU问题

51自学网 2023-06-16 18:55:52

python

这篇教程tensorflow之如何使用GPU而不是CPU问题写得很实用，希望能帮到您。

如何使用GPU而不是CPU

首先查看设备

from tensorflow.python.client import device_libprint(device_lib.list_local_devices())

如果发现只有一个CPU可用

则说明可能存在一下情况：

1 tensorflow-gpu是否安装，版本查看，如果版本低于tensorflow，则默认启动CPU（t-gpu版本最好高于t）

pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple/

2 CUDA初始化

在开始训练的前面输入如下：

# 使用CPU：import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu，不注释就是使用gpu# 使用GPU：import osos.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号

tensorflow使用GPU的设置方式

方法一

Config=tf.ConfigProto(allow_soft_placement=True) 
Django如何判断访问来源是PC端还是手机端
Django如何利用uwsgi和nginx修改代码自动重启

自学教程：tensorflow之如何使用GPU而不是CPU问题

目录

如何使用GPU而不是CPU

首先查看设备

如果发现只有一个CPU可用

tensorflow使用GPU的设置方式

方法一