天池实验室有60个小时的计算时长,对于小型项目而言是足够的。那么步骤应该选择什么?
首先,要新建一个Notebook。
其次,安装必要的包。
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118这样的话,能够避免因为各个包组合错误导致的问题。
第三,选择进入GPU独享型。
并且增加一个代码验证
nvidia -smi第四,开始进入检验
Python main.py
