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

标签 path pytorch

当我通过 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/

    相关文章:

    python - 如何修改此 PyTorch 卷积神经网络以接受 64 x 64 图像并正确输出预测?

    python - 在 Python 中查找数组中列表的元素(Numpy、Pytorch)

    html - SVG 中的中心路径

    java - 无法从 spring boot .jar 读取文件

    java - 尝试将文件路径包含到索引页时出现问题

    python - torch torch .load ModuleNotFoundError : No module named 'utils'

    python - 如何为 PyTorch 的 F.affine_grid 和 F.grid_sample 创建剪切矩阵?

    pytorch - 无法在 Azure Databricks GPU 中安装 Flash Attention(对于 Hugging Face 模型)

    java - 解析 Java 路径的安全方法

    c++ - 如何更改路径以修复错误“./main : error while loading shared library libmkl_core. 所以?