仅使用 Colab Pro 的 GPU

Posted 2023-03-16

【中文标题】仅使用 Colab Pro 的 GPU

【英文标题】：Just using the GPU from Colab Pro

【发布时间】：2021-08-07 06:02:10

【问题描述】：

今年夏天我正在研究一个研究项目。该项目是扩展机器学习程序的一些功能（位于 github）。我无法在笔记本电脑上训练模型，因为我只有 16 GB 的内存 (CPU)。这个项目是通过几个 python 文件和目录分派的。是否可以使用 Colab Pro 运行该程序而无需重写 Colab 单元中的 python 代码？我只需要 GPU。

【参考方案1】：

您可以执行以下操作：

将您的 Python 文件上传到 Google 云端硬盘 将 Google 云端硬盘装载到 Colab 将 Python 文件从 Google Drive 复制到 Colab 工作目录 在 Colab 中导入您的 Python 模块/包

您可以查看this answer了解更多详情。