- 博客(2)
- 收藏
- 关注
RSS订阅原创 解决方法：分布式训练卡住，GPU利用率100%
之后我尝试了很多方法依然存在卡住的问题，这个卡住和报错不一样，需要自己一点点去debug。后来，我仅仅使用单卡来跑模型训练，发现就不会卡住。所以基本上可以锁定是GPU之间通信的问题，可能是因为GPU之间的进程不同步导致的死锁。然后我在github上面找到有人说，把GPU的P2P通信的环境变量设置成0解决了：我尝试了之后发现不仅没解决，训练的速度还变慢了。所以只能继续debug，然后发现把整个模型evaluate部分的代码注释掉后，模型训练就不会卡住。
2025-06-17 15:08:53
821
原创 PCA 人脸识别 Python 附源码
本文采用orl4646的数据集，其中包括400张46*46的人脸照片，照片中一共有40个不同的人的人脸照片，每个人有10张人脸照片。本文采用PCA降维的主成分分析的方法，采用7：3的比例把数据分为训练集和验证集。把46*46的人脸图片降维到100维，通过比较验证集和训练集的余弦相似度，采用KNN的方法预测模型。有任何疑问欢迎和我交流。
2023-11-24 20:01:51
1700
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人