📝 主旨内容

环境准备

单GPU

代码内容

运行命令

• gpu：显卡编号

• size：矩阵大小，如果单卡20G显存，可以设置参数为37964，24G显存为39610，其他情况慢慢试

• minu：运行时长，单位-分钟

多GPU

代码内容

运行命令

• gpus：可用显卡数量

• size：矩阵大小，如果双卡32G显存，可以设置参数为30000，其他情况慢慢试

• minu：运行时长，单位-分钟

🤗 总结归纳

📎 参考文章

基于矩阵乘法的GPU烤机python代码（pytorch版）_用python语言编写gpu计算程序,实现矩阵的乘法计算。-CSDN博客

文章浏览阅读424次，点赞8次，收藏4次。基于矩阵乘法的GPU烤机python代码（pytorch版）_用python语言编写gpu计算程序,实现矩阵的乘法计算。

https://blog.csdn.net/qq_53644346/article/details/138508004