A error when using GPU

[janehu]

I’ve got the same error. even after I update CUDA to 10.
I happened to find a way to remove it. Now my code of training works.

• cuda: 10.0
• python: 3.7
• pytorch: 1.0
• cudnn: 7
• GPU: 2080ti

Invalid argument error with 2080Ti and cuda10, this removed but along with other error

I train my network and raise the error: THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument. Enviroment: docker: yes cuda: 10.0 python: 3.7 pytorch: 1.0 cudnn: 7 simple easy example: import torch from torchvision.models import vgg16 model = vgg16().cuda() x = torch.zeros((32, 3, 227, 227)).cuda() model(x) out: THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=11 : invalid argument tensor([[0., 0., 0., …, 0., 0., 0.]…

however, another problem came along when run with 'with torch.no_grad(), the output are all nans.
anyone know this?