美文网首页
torch.utils.cpp_extension.load

torch.utils.cpp_extension.load

作者: 承诺一时的华丽 | 来源:发表于2021-08-25 16:27 被阅读0次

    多进程对同一内容编译时,导致编译文件锁住了,清空缓存文件即可

     if version != old_version:
            print('build_directory lock:',build_directory)  # 打印缓存文件位置
            baton = FileBaton(os.path.join(build_directory, 'lock'))
            if baton.try_acquire():
                try:
    
    Windows
    缓存目录
    • 删除Cache目录即可

    C:\Users\win 10\AppData\Local\torch_extensions\torch_extensions\Cache

    Linux
    rm -rf /home/account_name/.cache/torch_extensions/*
    

    相关文章

      网友评论

          本文标题:torch.utils.cpp_extension.load

          本文链接:https://www.haomeiwen.com/subject/gvxeiltx.html