RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

最新推荐文章于 2023-08-08 22:57:04 发布

原创

最新推荐文章于 2023-08-08 22:57:04 发布 · 664 阅读

0 ·

CC 4.0 BY-SA版权

在尝试使用CUDA 11运行PyTorch时遇到了RuntimeError: cuDNN错误，状态未初始化。通过参考Stack Overflow上的解决方案，可以解决此问题。

cuda11运行pytorch报错

Traceback (most recent call last):
  File "train.py", line 220, in <module>
    main(args)
  File "train.py", line 89, in main
    train(model, args, train_loader, criterion, Hash_center, optimizer, epoch, two_loss_epoch)
  File "train.py", line 121, in train
    y = model(input)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 167, in forward
    outputs = self.parallel_apply(replicas, inputs, kwargs)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 177, in parallel_apply
    return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in p

最低0.47元/天解锁文章