如何修改火炬轮毂模型的下载路径

Question

当我通过火炬中心下载模型时，模型会自动下载到/home/me/.cache/torch。

如何修改此行为？

Answer 1

从官方文档中，有几种方法可以修改此路径。
按优先级顺序：

1. 调用 hub.set_dir((

2. $TORCH_HOME/hub，如果设置了环境变量TORCH_HOME。

3. $XDG_CACHE_HOME/torch/hub，如果设置了环境变量XDG_CACHE_HOME。

4. ~/.cache/torch/hub

所以我只需要做：

export TORCH_HUB=/my/path/

---

编辑

TORCH_HUB似乎已弃用，请改用TORCH_HOME