situ课题组工作站神经网络训练速度测评

2021/11/24 23:41:57

本文主要是介绍situ课题组工作站神经网络训练速度测评,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

situ课题组工作站神经网络训练速度测评
测试模型:Unet的一个语义分割模型,3000张图片(训练集2430张,验证集270张,测试集300张);
模型参考: U-Net编程实战——CT影像的肿瘤分割

  1. A100(计算能力8):运行一个epoch需要6s;
    在这里插入图片描述
  2. A6000(计算能力8.6):运行一个epoch需要9s;
    在这里插入图片描述
  3. RTX3090(计算能力8.5):运行一个epoch需要8s;
    在这里插入图片描述
  4. P6000(计算能力6.1):运行一个epoch需要18s;
    在这里插入图片描述
  5. GTX1080ti(计算能力6.1):运行一个epoch需要18s;
    在这里插入图片描述
    哦,yes,课题组新进一批顶级丹炉,训练速度比原来的1080ti提升了两倍以上,哈哈哈~。欢迎有志之士报考我们中国科学院上海光机所司徒老师的计算光学成像课题组,嘻嘻!


这篇关于situ课题组工作站神经网络训练速度测评的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程