CUDA: 9.0
CUDNN: 7.4
GPU: 2080ti
Python: 3.6
Installed pytorch via “pip3 install torch torchvision”
I have THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument error while run

import torch
from torchvision.models import vgg16
model =  vgg16().cuda()
x = torch.zeros((32, 3, 227, 227)).cuda()
model(x)

Crash while run first conv layer.
How to solve this?