pycharm - 在 pytorch 中使用卷积层时如何修复退出代码为 -1073741819 (0xC0000005) 的进程,向后错误()

标签 pycharm pytorch conv-neural-network cudnn

当我在 pytorch 上使用 Conv1dConv2d 层时,进程意外终止。我在以下行中收到错误:

loss.backward()

我的设置:

  • Windows 10
  • cuda 10.2
  • cudnn 7.6.5
  • RTX 2060 super
  • Nvidia 驱动程序 451.67
  • Pycharm 2020.04

错误:

Process finished with exit code -1073741819 (0xC0000005)

相比之下,当我将 conv 层替换为 dense 层时,问题不会出现。

为了进一步比较,相同的项目和相同的代码也在 Ubuntu 20.04 上运行,并且运行良好。

最佳答案

似乎有一个已知的bug在 GPU 上运行时,Windows 上的 Pytorch 会出现这个问题(使用 CUDA)。

确保提供给 Conv1dConv2d 的所有参数都是正确的,尤其是 padding 值。请注意,它可能与其他操作系统(如 linux/ubuntu)有不同的行为。

另外,如果您使用的是 Python-3.6 或更高版本,它可能是这个 bug .在那种情况下尝试使用 Python-3.5

关于pycharm - 在 pytorch 中使用卷积层时如何修复退出代码为 -1073741819 (0xC0000005) 的进程,向后错误(),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63059001/

相关文章:

python - Pycharm (Ubuntu) 程序无法打开

pycharm - 如何选择 PyCharm 中当前行下方的所有行

python - 请求模块在控制台中工作,但在脚本中不起作用 - 所有其他模块都正常工作

python - torch : "Model Weights not Changing"

machine-learning - 如何知道传递给GPU的数据是否会导致CUDA内存不足

python - 如何在 Tensorflow 中恢复训练好的模型并计算测试精度

python - 默认情况下如何打开交互式python控制台?

python - PyTorch:权重的 Sigmoid?

python - 如何在多核多线程上运行 TensorFlow

machine-learning - Keras/TensorFlow BatchNormalization 正在规范化什么