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

Question

当我通过 Torch Hub 下载模型时，模型会自动以/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改用