I cannot reproduce it on a V100-SMX2 32GB using PyTorch 1.6.0 and the CUDA10.1 binaries and get a valid output of torch.Size([64, 102, 2]).
EDIT: the only difference seems to be the driver, so could you update it?