RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383

[ouening]

OS: Ubuntu
pytorch: 1.1.0
cuda: 9.0
cudnn:7.6
GPU: RTX 2080

I got errors below:
runpy.run_path(target, run_name='__main__') File "/usr/lib/python3.6/runpy.py", line 263, in run_path pkg_name=pkg_name, script_name=fname) File "/usr/lib/python3.6/runpy.py", line 96, in _run_module_code mod_name, mod_spec, pkg_name, script_name) File "/usr/lib/python3.6/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/gaoya/Downloads/pytorch-cifar10-master/main.py", line 155, in <module> main() File "/home/gaoya/Downloads/pytorch-cifar10-master/main.py", line 27, in main solver.run() File "/home/gaoya/Downloads/pytorch-cifar10-master/main.py", line 145, in run train_result = self.train() File "/home/gaoya/Downloads/pytorch-cifar10-master/main.py", line 95, in train output = self.model(data) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "/home/gaoya/Downloads/pytorch-cifar10-master/models/AlexNet.py", line 38, in forward x = self.features(x) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/container.py", line 92, in forward input = module(input) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 338, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383