利用天池实验室GPU资源跑通代码

天池实验室有60个小时的计算时长,对于小型项目而言是足够的。那么步骤应该选择什么?

首先,要新建一个Notebook。

其次,安装必要的包。

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

这样的话,能够避免因为各个包组合错误导致的问题。

第三,选择进入GPU独享型。

并且增加一个代码验证

nvidia -smi

第四,开始进入检验

Python main.py

发表评论

您的邮箱地址不会被公开。 必填项已用 * 标注