-
Notifications
You must be signed in to change notification settings - Fork 194
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
gpu benchmark does not support PyTorch 1.5.0. #77
Comments
现在比较的是1.4.0。已知Pytorch 1.5.0对tensor transpose,concat的操作和1.4.0结果不一致,这是Pytorch bug还是feature有待考证。 |
谢谢!测了一下,pytorch1.4.0和1.5.0在该GPU benchmark上确实没有什么性能差别。 |
The member function weight = torch.clone(torch.t(pooler_params['dense.weight'])) I have no idea, why it does not work as predicted in Pytorch 1.5.0. |
The bug will fix in version v0.3.0 |
The bug fixed! |
Great work! |
您好,我试图用最新的pytorch做gpu benchmark的对照组,配置如下:
pytorch: 1.5.0
torchvision: 0.6.0
CUDA: 10.2
OS: Ubuntu18.04
即Dockerfile.gpu里对应行修改为"conda install pytorch=1.5.0 torchvision=0.6.0 cudatoolkit=10.2 -c pytorch"
在docker里build后在test时,有好几个测试用例通不过:
...
请问目前这个gpu benchmark对pytorch版本最高支持到多少呢?官方首页里的benchmark实验结果,是和哪个版本的pytorch比较的呢?
The text was updated successfully, but these errors were encountered: