the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653215914/work/aten/src/THC/THCBlas.cu:259

[szpxmu]

warnings.warn('You have chosen a specific GPU. This will completely ' Use GPU: 0 for training Loaded pretrained weights for efficientnet-b3 => using pre-trained model 'efficientnet-b3' Using image size 300 Traceback (most recent call last): File "/home/szp/EfficientNet-PyTorch-master/examples/imagenet/main.py", line 443, in main() File "/home/szp/EfficientNet-PyTorch-master/examples/imagenet/main.py", line 117, in main main_worker(args.gpu, ngpus_per_node, args) File "/home/szp/EfficientNet-PyTorch-master/examples/imagenet/main.py", line 254, in main_worker res = validate(val_loader, model, criterion, args) File "/home/szp/EfficientNet-PyTorch-master/examples/imagenet/main.py", line 348, in validate output = model(images) File "/home/szp/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, **kwargs) File "/home/szp/EfficientNet-PyTorch-master/efficientnet_pytorch/model.py", line 320, in forward x = self._fc(x) File "/home/szp/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, **kwargs) File "/home/szp/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 92, in forward return F.linear(input, self.weight, self.bias) File "/home/szp/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/functional.py", line 1406, in linear ret = torch.addmm(bias, input, weight.t()) RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653215914/work/aten/src/THC/THCBlas.cu:259

Process finished with exit code 1