RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

原创 于 2020-12-24 21:20:46 发布 · 305 阅读 · 0 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

该博客讨论了如何在使用 PyTorch 的深度学习项目中禁用 CUDA 和 CUDNN，以提高稳定性和控制计算速度。通过设置 `torch.backends.cudnn.benchmark=False` 和 `torch.backends.cudnn.enabled=False`，可以避免由于硬件不一致导致的性能波动，确保实验的可重复性。

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

torch.backends.cudnn.benchmark = False
torch.backends.cudnn.enabled = False

评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值