- 博客(2)
- 收藏
- 关注
RSS订阅原创 AutoDL 跑深度学习报错：RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
但想来想去，Auto DL平台CUDA与cuDNN是配置好的啊，不会出现CUDA和cuDNN版本与PyTorch不兼容的问题。也有伙伴说是由于显存不足，需要调整Batchsize，于是把Batchsize从32依次调整到16、8、4，仍然不管用，且运行。在多GPU环境下，有时候默认情况下程序可能会选择不合适的GPU，导致出现一些问题，比如显存溢出或者资源冲突等。环境变量，可以显式地告诉程序在哪块GPU上运行，从而避免这些问题。发现显存是没问题的，后来在运行之前加入了下面这行代码。# 创建输入数据和网络。
2024-04-07 10:01:52
728
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人