当前位置:网站首页>RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
2022-07-06 09:16:00 【曹家小圆宝】
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 parallel_apply
output.reraise()
File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/_utils.py", line 429, in reraise
raise self.exc_type(msg)
RuntimeError: Caught RuntimeError in replica 0 on device 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker
output = module(*input, **kwargs)
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 "/mnt/caojie/workspace/Hadamard-Matrix-for-hashing/network.py", line 35, in forward
x = self.feature_layers(x)
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/modules/container.py", line 119, in forward
input = module(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/modules/conv.py", line 399, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 396, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
解决方法:
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
边栏推荐
- There are three iPhone se 2022 models in the Eurasian Economic Commission database
- 5G工作原理详解(解释&图解)
- imgcat使用心得
- RT-Thread 线程的时间片轮询调度
- C language callback function [C language]
- STM32 如何定位导致发生 hard fault 的代码段
- Yarn installation and use
- nodejs连接Mysql
- Some concepts often asked in database interview
- Stage 4 MySQL database
猜你喜欢
随机推荐
Pytorch实现简单线性回归Demo
Composition des mots (sous - total)
MongoDB
STM32型号与Contex m对应关系
关键字 inline (内联函数)用法解析【C语言】
Reno7 60W super flash charging architecture
Analysis of charging architecture of glory magic 3pro
共用体(union)详解【C语言】
I2C bus timing explanation
电商数据分析--薪资预测(线性回归)
uCOS-III 的特点、任务状态、启动
Detailed explanation of 5g working principle (explanation & illustration)
ESP8266使用arduino连接阿里云物联网
Word排版(小計)
Come and walk into the JVM
互聯網協議詳解
C language callback function [C language]
物联网系统框架学习
高通&MTK&麒麟 手机平台USB3.0方案对比
imgcat使用心得