谷歌协作本地运行使用本地GPU

Question

我使用collaboratic和Pytorch来运行一个使用不寻常数据集的GAN，该数据集目前存储在我的机器上。为了访问这些文件，我连接到一个本地运行时（根据https://research.google.com/colaboratory/local-runtimes.html）。然而，colaboratorial现在在运行时使用了我自己的GPU，这在以前的运行中没有做到。我知道这是因为目前的运行速度慢得多，因为他们使用的是我的GTX 1060 6GB，而不是Colab的特斯拉K80。在

我用

torch.cuda.get_device_name(0)

当我在本地连接时，返回“GeForce GTX 1060 6G”。即使选择了编辑->笔记本设置->硬件加速器->GPU”，也是如此。在

但是，如果我没有本地连接，而是使用（默认）“连接到托管运行时”选项

torch.cuda.get_device_name(0)不返回“Tesla K80”。在

我在将数据集上载到驱动器时遇到了问题，因为它是一个大的图像数据集，我想继续使用本地运行时。在

如何同时使用本地运行时和Colab的惊人特斯拉K80？ 任何帮助都将不胜感激。在

Answer 1

This回答您的问题。检查连接。在