其他分享
首页 > 其他分享> > situ课题组工作站神经网络训练速度测评

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提升了两倍以上,哈哈哈~。欢迎有志之士报考我们中国科学院上海光机所司徒老师的计算光学成像课题组,嘻嘻!

标签:课题组,训练,epoch,神经网络,6.1,计算能力,situ,运行
来源: https://blog.csdn.net/qq_34917728/article/details/121527556