首页 > 解决方案 > 在 VScode 中使用 jupyter note 时命令未运行

问题描述

我正在使用 VScode 中支持的 Jupyter notebook 来开发图像分类模型。

问题是在 Colab 或 jupyter 中运行良好的命令在 VScode 上的 Jupyter 笔记本中运行不佳。

如下图所示, 在此处输入图片描述 在 Colab 中运行训练命令时,运行时间约为 10 分钟。

在此处输入图像描述 但是,在 VScode 上的 Jupyter 笔记本中,在我的服务器上运行大约 8.6 秒后,它被检查为是否正常工作,然后完成。

这时候通过nvidia-smi观察使用情况。在 CPU 的情况下,使用率上升到 135%,然后在完成阶段下降,也就是 8.6 秒后。GPU似乎根本不工作。

我在这里想知道的是,

  1. VScode中的Jupyter notebook有问题吗(因为在colab中可以正常工作)
  2. 或者是我服务器的深度学习GPU设置有问题(因为我不使用GPU,它只使用CPU然后关闭)

如果您是专家,请帮助

我的服务器规格:

该命令是从该站点引用的: https ://learnopencv.com/classification-with-localization/

标签: pythontensorflowvisual-studio-codejupyter-notebookgpu

解决方案


推荐阅读