search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.55k stars 419 forks source link

RuntimeError:CUDNN_Status_Mapping_Errror #121

Open bheAI opened 4 years ago

bheAI commented 4 years ago

I run in multi (2~4) GPUs (Tesla V100), there are something error.

Found GPU0 V100-SXM2-16GB which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time, but your PyTorch was compiled with CUDA_VERSION 8000. Please install the correct PyTorch binary using instructions from http://pytorch.org warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION)) ...... ../python3.5/dist-packages/torch/nn/modules/module.py, line 491, incall result = self.forward(input, *kwargs) ......

RuntimeError:CUDNN_Status_Mapping_Errror

JCZ404 commented 3 years ago

Hi, bro., I also meet the same problem, have you already fixed it?