修改 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