Hi all - I started experiencing this issue recently on a new set of A100 GPUs. I am using the following configuration:
NVIDIA A100 GPU
torch == 1.4.0
Python == 3.7.4-GCCcore-8.3.0
CUDA == 10.1.243-GCC-8.3.0
cudnn == 7.6.5.32-10.1
The error is occurring when I am calling forward on a sequence of convolutional operations in my code.
I have used the same configuration on other GPUs, such as K80, P100, V100 and RTX8000, and never had the error. Anybody found a solution to this issue?