Error Message: /pytorch/aten/src/THC/THCBlas.cu:206

1

Hi, I am getting the following error message while performing eval testing but my code still runs without stopping. My CUDA version is already 9.2. Just wondering why prompted this error? My pytorch version is 1.3.0 cu92

/pytorch/aten/src/THC/THCBlas.cu:206: UserWarning: Matrix multiplication for dimensions larger than 2^21 has known bugs on your combination of CUDA version and device type. Please consider upgrading to CUDA 9.2 or later.

2

What does torch.cuda.version() print?

3

Hi,
I can’t run torch.cuda.version() as it throws this error AttributeError: module 'torch.cuda' has no attribute 'version'. Not sure if torch.version.cuda is the one you after but this is the output: '9.2.148'

4

Yes, sorry for the wrong command.
OK, in that case you could probably ignore this warning.
I’ll have a look, why the warning is thrown even though your CUDA version shouldn’t trigger it.

Thanks for raising this issue :slight_smile: