torch.backends.cudnn.benchmark = False
torch.backends.cudnn.enabled = False
RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405
最新推荐文章于 2024-09-27 16:46:47 发布
该博客讨论了如何在使用 PyTorch 的深度学习项目中禁用 CUDA 和 CUDNN,以提高稳定性和控制计算速度。通过设置 `torch.backends.cudnn.benchmark=False` 和 `torch.backends.cudnn.enabled=False`,可以避免由于硬件不一致导致的性能波动,确保实验的可重复性。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4879

被折叠的 条评论
为什么被折叠?



