何喆379 2022-03-02 23:40 采纳率: 100%
浏览 47
已结题

jupyter 中tf 搭建神经网络 如何确定自己的GPU参与了计算?

采用mnist的数据集练习 神经网络分类问题,搭建完模型后的fit过程如下:

%%time
model.fit(x_train_reshaped,y_train,epochs = 10,batch_size = 64,
          validation_data = (x_valid_reshaped,y_valid))
# 这里面一定要用reshaped的数据才行

运算结果如下:

Epoch 1/10
938/938 [==============================] - 4s 4ms/step - loss: 3.0898 - sparse_categorical_accuracy: 0.6319 - val_loss: 0.7920 - val_sparse_categorical_accuracy: 0.7922
Epoch 2/10
938/938 [==============================] - 3s 4ms/step - loss: 0.6585 - sparse_categorical_accuracy: 0.8172 - val_loss: 0.5834 - val_sparse_categorical_accuracy: 0.8415
Epoch 3/10
938/938 [==============================] - 3s 4ms/step - loss: 0.5044 - sparse_categorical_accuracy: 0.8641 - val_loss: 0.4774 - val_sparse_categorical_accuracy: 0.8803
Epoch 4/10
938/938 [==============================] - 3s 4ms/step - loss: 0.4253 - sparse_categorical_accuracy: 0.8852 - val_loss: 0.3929 - val_sparse_categorical_accuracy: 0.8931
Epoch 5/10
938/938 [==============================] - 4s 4ms/step - loss: 0.3789 - sparse_categorical_accuracy: 0.8986 - val_loss: 0.3681 - val_sparse_categorical_accuracy: 0.9028
Epoch 6/10
938/938 [==============================] - 4s 4ms/step - loss: 0.3455 - sparse_categorical_accuracy: 0.9064 - val_loss: 0.3584 - val_sparse_categorical_accuracy: 0.9103
Epoch 7/10
938/938 [==============================] - 3s 4ms/step - loss: 0.3146 - sparse_categorical_accuracy: 0.9131 - val_loss: 0.3123 - val_sparse_categorical_accuracy: 0.9160
Epoch 8/10
938/938 [==============================] - 3s 4ms/step - loss: 0.2925 - sparse_categorical_accuracy: 0.9198 - val_loss: 0.3075 - val_sparse_categorical_accuracy: 0.9169
Epoch 9/10
938/938 [==============================] - 4s 4ms/step - loss: 0.2782 - sparse_categorical_accuracy: 0.9214 - val_loss: 0.3440 - val_sparse_categorical_accuracy: 0.9157
Epoch 10/10
938/938 [==============================] - 4s 4ms/step - loss: 0.2652 - sparse_categorical_accuracy: 0.9247 - val_loss: 0.3154 - val_sparse_categorical_accuracy: 0.9147
Wall time: 35.9 s

我想问在这个过程中GPU 有没有参与计算? 如果没有,怎样设置才可以GPU 参与计算,在这之前我也将CUDA 显卡驱动什么的都装好了。
数据量测试集是 60000784 验证集是10000784.

  • 写回答

1条回答 默认 最新

  • ·星辰大海 2022-03-03 10:17
    关注

    有个简单的办法,提前打开任务管理器翻到GPU那一栏,在训练或者预测数据的时候观察各项指标特别是复制有没有起伏

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

    报告相同问题?

    问题事件

    • 系统已结题 3月11日
    • 已采纳回答 3月3日
    • 创建了问题 3月2日

    悬赏问题

    • ¥15 Tpad api账户 api口令
    • ¥30 ppt进度条制作,vba语言
    • ¥15 stc12c5a60s2单片机测光敏ADC
    • ¥15 生信simpleaffy包下载
    • ¥15 请教一下simulink中S函数相关问题
    • ¥15 在二层网络中,掩码存在包含关系即可通信
    • ¥15 端口转发器解析失败不知道电脑设置了啥
    • ¥15 Latex算法流程图行号自定义
    • ¥15 关于#python#的问题:我在自己的电脑上运行起来总是报错,希望能给我一个详细的教程,(开发工具-github)
    • ¥40 基于51单片机实现球赛计分器功能