在发生错误,报CUDA的assert,比如什么index out of range的问题。
那么可能因为并行导致报错不一致,甚至报错的未知都不对,这时候需要在程序中加入以下的代码:
import os
os.environ[‘CUDA_LAUNCN_BLOCKING’]=‘1’
在发生错误,报CUDA的assert,比如什么index out of range的问题。
那么可能因为并行导致报错不一致,甚至报错的未知都不对,这时候需要在程序中加入以下的代码:
import os
os.environ[‘CUDA_LAUNCN_BLOCKING’]=‘1’
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4179
3128
1496

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