量子位
25-11-17 11:31 微博认证:量子位官方微博

#用VSCode白嫖GPU##VSCode能连Colab了#

Google Colab现在可以直接在VS Code里跑了,连上GPU/TPU只要几步,全流程免费!

Colab原本是一个网页端的Jupyter平台,深度学习开发者都很熟,最大优势就是能免费用上Google提供的GPU和TPU算力。但之前用它只能靠浏览器,不太适合做大项目。

现在好了,官方出了VS Code扩展插件,直接把Colab运行时接进本地VS Code编辑器里,不用离开熟悉的工作环境,想写代码就写,想训练模型就连云端资源。

比如在视频里,作者打开了一个讲Pandas的数据分析Notebook,几秒就跑起来了。包括`import numpy`、创建Pandas对象这些常见操作,直接在Colab云端算力上执行,速度也快。

用法非常简单,几步就能连上:

1. 打开VS Code扩展市场,搜“Google Colab”并安装;

2. 打开本地`.ipynb`文件;

3. 点击右上角选择内核,选“Colab”;

4. 登录Google账号,挑个云端运行环境(可以选GPU/TPU);

5. 开干。

整个插件是免费的,而且设置只需要2分钟,适合想快速搭GPU训练环境的AI/ML项目开发者。

感兴趣的小伙伴可以点击:developers.googleblog.com/en/google-colab-is-coming-to-vs-code/ http://t.cn/AX21ckgp