【发布时间】:2020-02-25 18:55:29
【问题描述】:
我正在试用 Google Colab,想知道我是否能够使用本地 CPU、RAM、SSD 和 GPU?我试图在我的 SSD 上搜索一个目录,但结果是空的。我发现我必须将我的目录上传到我的 Google Drive 并运行代码:
from google.colab import drive
drive.mount('/content/drive')
现在我已经建立了我的目录并运行了我的 UNet,我收到了警告 WARNING:tensorflow:There are non-GPU devices in `tf.distribute.Strategy`, not using nccl allreduce. 和 INFO:tensorflow:Using MirroredStrategy with devices ('/job:localhost/replica:0/task:0/device:CPU:0',)
我正在使用 Linux RedHat 8 并安装了 Nvidia NCCL 驱动程序。使用 Jupyter Notebook 时,我没有收到此警告。
在我的第一个 epoch 开始之前,我的会话崩溃并给出消息 Your session crashed after using all available RAM. Get more RAM。 Colab 在右上角显示 RAM:12.72 GB 和磁盘:107.77。
我的系统有一个用于存储的 TB SSD 和 128 GB 的 RAM。有没有办法使用我的资源运行 Colab,还是我必须使用他们的资源?
【问题讨论】:
-
使用你的存在带宽问题
标签: python tensorflow machine-learning google-colaboratory