Torch.prod produces RuntimeError: CUDA driver error: invalid argument

5

Hi @AlphaBetaGamma96 ,

I created two conda environments related to this issue, and torch.version.cuda s are ‘11.7’ and ‘11.3’. And they both have the same problem.

Edit: I’ve just tried Pytorch 2.0 with Cuda 11.8 and got the same error.