- 可以使用torch.utils.data.random_split()划分数据集,比赛中常用。不需要自己划分了
- 没有GPU的可以使用kaggle强行训练一波,就是kaggle有时间限制,并且需要打包上传代码和数据。(不太推荐,适合折磨心态)可以参考这里
- 使用vscode调试pytorch代码
- 使用
gpustat
查看谁在和你抢卡(用户和内存占用),嘿嘿。使用pip install gpustat
安装。watch --color -n1 gpustat -cpu
运行每1秒刷新一次gpu状态。 - 根据
PID
查询运行该线程的命令 :ps -o cmd fp <PID>
,这里<PID>
替换为数字,比如35000
记录一些深度学习中的技巧(持续更新)
于 2022-03-08 21:15:11 首次发布