设置cuda——launch——blocking

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