我在尝试使用Google Colab,想知道是否可以使用我的本地CPU、内存、SSD和GPU?我尝试搜索SSD上的一个目录,但结果为空。我发现我必须将我的目录上传到Google Drive并运行代码:
from google.colab import drivedrive.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时,我没有收到这个警告。
在我的第一个周期开始之前,我的会话崩溃并显示消息 Your session crashed after using all available RAM. Get more RAM
。在Colab的右上角显示RAM:12.72 GB,磁盘:107.77 GB。我的系统有一个1TB的SSD用于存储和128 GB的内存。有没有办法使用我的资源运行Colab,还是我必须使用他们的资源?
回答:
您可以通过选择“本地运行时”来使用您自己的本地资源。
请查看此处的文档:
https://research.google.com/colaboratory/local-runtimes.html