Google Colaboratory本地运行使用本地GPU

Question

Google Colaboratory本地运行使用本地GPU

pythonmachine-learningdeep-learningpytorchgoogle-colaboratory

13

我正在使用Colaboratory和PyTorch运行GAN，使用的是一组不寻常的数据集，这些数据集当前存储在我的本地机器上。为了访问这些文件，我连接到了一个本地运行时（根据https://research.google.com/colaboratory/local-runtimes.html）。然而，现在Colaboratory在运行时使用了我的GPU，而之前则没有这样做。我知道这是因为当前的运行速度非常慢，因为它们正在使用我的GTX 1060 6GB而不是Colab的Tesla K80。

我使用以下方式进行了检查：

torch.cuda.get_device_name(0)

当我本地连接时，它返回“GeForce GTX 1060 6G”。即使选择了“GPU”，这种情况仍然存在。但是，当我没有本地连接，而是使用（默认的）“连接到托管运行时”选项时，torch.cuda.get_device_name(0)返回“Tesla K80”。由于我的数据集是一个大型图像数据集，因此我上传到Drive遇到了麻烦，并希望继续使用本地运行时。如何同时使用本地运行时和Colab的惊人Tesla K80？任何帮助将不胜感激。

- krustybek

2

这是你的答案：https://dev59.com/GbDla4cB1Zd3GeqP7F0b - manza

@manza，请考虑将您的评论添加为答案，因为它似乎能够满足问题。 - Shagun Sodhani

这个回答解决了你的问题吗？Jupyter Notebook - GPU - Guglie

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- JaonHax · Accepted Answer

Colaboratory正在使用您的GPU，因为您将其连接到本地运行时。这就是将其连接到自己的运行时所意味的。这意味着您正在使用您自己的计算机来处理过程，而不是使用Google的服务器。如果您仍想使用Google的服务器和处理能力，我建议您研究一下将您的Google Drive连接到Colaboratory运行时。