gpt4 book ai didi

path - 如何修改 Torch Hub 模型的下载路径

转载 作者:行者123 更新时间:2023-12-03 14:16:12 25 4
gpt4 key购买 nike

当我通过 Torch Hub 下载模型时,模型会自动下载到 /home/me/.cache/torch .

如何修改此行为?

最佳答案

来自 official documentation ,有几种方法可以修改此路径。
按优先顺序:

  • 调用 hub.set_dir()
  • $TORCH_HOME/hub,如果设置了环境变量 TORCH_HOME。
  • $XDG_CACHE_HOME/torch/hub,如果设置了环境变量 XDG_CACHE_HOME。
  • ~/.cache/torch/hub

  • 所以我只需要这样做:
    export TORCH_HUB=/my/path/
    编辑
    TORCH_HUB似乎已弃用,请使用 TORCH_HOME反而

    关于path - 如何修改 Torch Hub 模型的下载路径,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59134499/

    25 4 0
    Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
    广告合作:1813099741@qq.com 6ren.com