Cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450

1

I was running a python script when I got this error: “RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:450”.

How can I solve this problem?

Pytorch version: 1.1.0
Cuda version: 9.0

Please let me know if any other details are required.

2

Could you update PyTorch to the latest release and check, if you are still seeing the issue, please?

3

Actually the script only works on 1.1.0. If I run it on the latest release I get the following error: “AttributeError: ‘AvgPool3d’ object has no attribute ‘divisor_override’”

4

The divisor_override argument isn’t available in nn.AvgPool3d in PyTorch 1.1.0 and was added in 1.2.0, so I’m unsure why your code snippet would try to call it after you’ve updated the PyTorch version.
Could you check what’s causing the AttributeError?