设置cuda——launch——blocking

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
posted @ 2024-06-28 16:17  祥瑞哈哈哈  阅读(140)  评论(0)    收藏  举报