pytorchhub解决bug：.cache\torch\hub\master.zip.....'另一个程序正在使用此文件,进程无法访问。'...

转载于 2019-06-14 01:56:00 发布 · 853 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/gambler/p/11020649.html

文章标签：

#python #人工智能 #开发工具

本文介绍了一个在尝试加载PyTorch Hub中pytorch/vision模型列表时遇到的Bug及其解决方法。通过临时修改hub.py文件中的_remove_if_exists()方法，成功解决了问题并获取了模型列表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

出现上述bug是在第一次查看pytorchhub可用模型的时候发生的

>>torch.hub.list('pytorch/vision')

没有记录下来，所以没办法重现

解决方法是：

定位到hub.py文件的_remove_if_exists(path)方法，pycharm定位方式是直接点击bug最后一行显示bug提示

例如：

该方法原代码如下：

def _remove_if_exists(path):
    if os.path.exists(path):
        if os.path.isfile(path):
            os.remove(path)
        else:
            shutil.rmtree(path)

修改为：

def _remove_if_exists(path):
    if os.path.exists(path):
        if os.path.isfile(path):
           # os.remove(path)
            pass
        else:
            shutil.rmtree(path)

然后执行

>>torch.hub.list('pytorch/vision')，发现能够成功获取所有模型