search

xiaoyufenfei / Efficient-Segmentation-Networks

Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)
MIT License
895 stars 163 forks source link

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED #27

Open GREENtt opened 1 year ago

GREENtt commented 1 year ago

Traceback (most recent call last): File "D:/Code/CodePycharm/Deep_Learning/segment/Efficient-Segmentation-Networks-master/train.py", line 409, in train_model(args) File "D:/Code/CodePycharm/Deep_Learning/segment/Efficient-Segmentation-Networks-master/train.py", line 222, in train_model lossTr, lr = train(args, trainLoader, model, criteria, optimizer, epoch) File "D:/Code/CodePycharm/Deep_Learning/segment/Efficient-Segmentation-Networks-master/train.py", line 336, in train loss.backward() File "D:\Environment\Anaconda\envs\lmffnet\lib\site-packages\torch\tensor.py", line 118, in backward torch.autograd.backward(self, gradient, retain_graph, create_graph) File "D:\Environment\Anaconda\envs\lmffnet\lib\site-packages\torch\autograd__init__.py", line 93, in backward allow_unreachable=True) # allow_unreachable flag RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

Process finished with exit code 1

can you help me? please.

zhiwenj-HNU commented 1 month ago

cuda version is incorrect. You need to reinstall CUDA 9.0 or 9.2

zhiwenj-HNU commented 1 month ago

cuda version is incorrect. You need to reinstall CUDA 9.0 or 9.2