ncu

kolaseen6 天前
pytorch·python·深度学习·docker·gpu·cuda·ncu
NCU使用指南及模型性能测试（pytorch2.5.1）本项目在原项目的基础上增加了Nsight Compute(ncu)测试的功能，并对相关脚本功能做了一些健硕性的增强，同时，对一些框架的代码进行了更改（主要是数据集的大小和epoch等），增加模型性能测试的效率，同时完善了模型LSTM的有关功能。