C盘是公共厕所吗？怎么谁都要进去拉屎？

本文章将介绍如何更改Torch和HuggingFace的下载/缓存目录

修改 torch 缓存路径

torch 使用 TORCH_HOME 环境变量来指定缓存目录。

Windows

设置环境变量

  • 右键点击“此电脑”，选择“属性”。
  • 点击“高级系统设置”。
  • 在“系统属性”窗口中，点击“环境变量”按钮。
  • 在“系统变量”下，点击“新建”按钮。
  • 设置变量名为 TORCH_HOME，变量值为你希望存储模型的路径（例如，D:\torch_cache）。

或者在命令行中设置

set TORCH_HOME=D:\torch_cache

Linux/MacOS

编辑 shell 配置文件（如 .bashrc, .zshrc）

export TORCH_HOME=/path/to/your/desired/cache/dir

应用更改

source ~/.bashrc  # or ~/.zshrc

修改 Huggingface Transformers 缓存路径

huggingface 使用 HF_HOME 或 TRANSFORMERS_CACHE 环境变量来指定缓存目录。

Windows

设置环境变量

  • 重复上述步骤，添加新的系统变量。
  • 设置变量名为 HF_HOME 或 TRANSFORMERS_CACHE，变量值为你希望存储模型的路径（例如，D:\huggingface_cache）。

或者在命令行中设置

set HF_HOME=D:\huggingface_cache

Linux/MacOS

编辑 shell 配置文件（如 .bashrc, .zshrc）

export HF_HOME=/path/to/your/desired/cache/dir
# 或者
export TRANSFORMERS_CACHE=/path/to/your/desired/cache/dir

应用更改

source ~/.bashrc  # or ~/.zshrc

移动现有的缓存文件

关闭正在运行的 Python 程序

移动现有的 .cache 文件夹到新的路径

  • 在 Windows 下，可以使用文件资源管理器进行移动。

  • 在 Linux/MacOS 下，可以使用 mv 命令：

    mv ~/.cache/torch /path/to/your/desired/cache/dir
mv ~/.cache/huggingface /path/to/your/desired/cache/dir

创建符号链接（可选）

  • 在 Linux/MacOS 下，可以创建符号链接以保持旧路径的可用性：

    ln -s /path/to/your/desired/cache/dir/torch ~/.cache/torch
ln -s /path/to/your/desired/cache/dir/huggingface ~/.cache/huggingface

标签

  • Windows
  • Linux