situ课题组工作站神经网络训练速度测评
作者:互联网
测试模型:Unet的一个语义分割模型,3000张图片(训练集2430张,验证集270张,测试集300张);
模型参考: U-Net编程实战——CT影像的肿瘤分割
- A100(计算能力8):运行一个epoch需要6s;
- A6000(计算能力8.6):运行一个epoch需要9s;
- RTX3090(计算能力8.5):运行一个epoch需要8s;
- P6000(计算能力6.1):运行一个epoch需要18s;
- GTX1080ti(计算能力6.1):运行一个epoch需要18s;
哦,yes,课题组新进一批顶级丹炉,训练速度比原来的1080ti提升了两倍以上,哈哈哈~。欢迎有志之士报考我们中国科学院上海光机所司徒老师的计算光学成像课题组,嘻嘻!
标签:课题组,训练,epoch,神经网络,6.1,计算能力,situ,运行 来源: https://blog.csdn.net/qq_34917728/article/details/121527556