RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

最新推荐文章于 2023-09-02 17:53:09 发布

曹家小圆宝

最新推荐文章于 2023-09-02 17:53:09 发布

阅读量635

点赞数

分类专栏： pytorch cuda

本文链接：https://blog.youkuaiyun.com/fighting_Kitty/article/details/116500976

版权

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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/s

最低0.47元/天解锁文章