params = {
"model": "UNet11",
"device": "cuda",
"lr": 0.001,
"batch_size": 16,
"num_workers": 4,
"epochs": 10,
}
model = create_model(params)
将num_workers改为0
如果还是没有解决，试试将batchsize设置小一点。
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
2021-11-19
984
版权
版权声明：
本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《
阿里云开发者社区用户服务协议》和
《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写
侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
简介：
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
目录
相关文章
|
IDE
PyTorch
网络安全
|
缓存
PHP
Composer 报错 Error while processing content unencoding: Unknown failure within ...
Composer 报错 Error while processing content unencoding: Unknown failure within ...
193
0
0
|
1月前
|
并行计算
PyTorch
Linux
【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
本文提供了在使用PyTorch时遇到cuDNN error: CUDNN_STATUS_NOT_INITIALIZED错误的几种解决方法，包括重新安装PyTorch及其相关库以确保版本兼容，检查CUDA与显卡驱动的对应关系，以及在无法使用GPU的情况下切换到CPU运行模型。
292
2
2
|
3月前
|
Linux
Go
C语言
【ERROR】chaincode install failed with status: 500 - failed to invoke backing implementation xxx
【ERROR】chaincode install failed with status: 500 - failed to invoke backing implementation xxx
57
0
0
|
4月前
|
机器学习/深度学习
并行计算
PyTorch
【已解决】RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors might be asynchronous
【已解决】RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors might be asynchronous
745
2
2
|
4月前
|
并行计算
PyTorch
算法框架/工具
【已解决】RuntimeError: CuDA error: no kernel image is available for execution on the device
【已解决】RuntimeError: CuDA error: no kernel image is available for execution on the device
933
1
1
|
C语言
安装mujoco报错：distutils.errors.DistutilsExecError: command ‘gcc‘ failed with exit status 1
安装mujoco报错：distutils.errors.DistutilsExecError: command ‘gcc‘ failed with exit status 1
571
0
0
|
Unix
Linux
异构计算
成功解决 ERROR: An error occurred while performing the step: “Building kernel modules“. See /var/log/nv
成功解决 ERROR: An error occurred while performing the step: “Building kernel modules“. See /var/log/nv
4385
0
0
解决 RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
解决 RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
300
0
0
|
异构计算
GPU报错：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
832
0
1