是否可以使用GPU加速在本地运行.ipynb笔记本？多么?

Question

每次我需要训练一个‘大型’深度学习模式，我从谷歌Collab，因为它允许你使用GPU加速。

我的电脑有一个专用的GPU，我想知道是否可以用它快速地在本地运行我的笔记本电脑。可以用我的个人电脑GPU训练模型吗？在这种情况下，怎么做？

我愿意使用DataSpell、VSCode或任何其他IDE。

Answer 1

尼古拉斯·雷诺特( Nicholas )有一个很棒的“入门”视频，它贯穿于在你的个人电脑上安装GPU加速笔记本的整个过程。你感兴趣的东西从12分钟左右开始。

Answer 2

是的，使用GPU加速在本地运行.ipynb笔记本是可能的。为此，您需要安装必要的库和框架，如TensorFlow、PyTorch或Keras。根据您选择的IDE，您需要安装GPU加速的相关插件和软件包。

在IDE方面，DataSpell、VSCode、PyCharm和木星笔记本都适合在本地运行GPU加速的笔记本。

一旦安装了必要的库和框架，您就需要为GPU安装适当的驱动程序，并为GPU加速配置环境。

最后，您需要修改.ipynb笔记本以启用GPU加速，并指定将要使用的GPU数量。一旦采取了所有必要的步骤，你将能够运行本地的GPU加速笔记本。