如何在Google Colab中将Huggingface缓存目录迁移至Google Drive？

• 挂载Google Drive
  这是基础前提，先执行以下代码完成挂载授权：

  from google.colab import drive
  drive.mount('/content/drive')
  

• 创建目标缓存目录
  手动创建文件夹避免路径不存在的问题：

  !mkdir -p '/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash'
  !mkdir -p '/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash/Datasets'
  

• 设置环境变量（二选一即可）
  之前的export无效是因为没传递给Python进程，试试这两种可靠方式：

  • Python内直接设置（推荐）

    import os
    os.environ['TRANSFORMERS_CACHE'] = '/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash'
    os.environ['HF_DATASETS_CACHE'] = '/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash/Datasets'
    os.environ['HF_HUB_CACHE'] = '/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash'
    

  • Colab魔法命令设置

    %env TRANSFORMERS_CACHE=/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash
    %env HF_DATASETS_CACHE=/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash/Datasets
    %env HF_HUB_CACHE=/content/drive/MyDrive/Colab Notebooks/NLP/HuggingfaceCash
    

• 验证缓存路径
  执行代码确认设置成功：

  import os
  from huggingface_hub import get_cache_dir
  
  print("Transformers缓存路径:", os.environ.get('TRANSFORMERS_CACHE'))
  print("Datasets缓存路径:", os.environ.get('HF_DATASETS_CACHE'))
  print("HF Hub缓存路径:", get_cache_dir())
  

• 常见问题排查

  • 仍使用默认缓存：检查GDrive挂载状态、目标目录是否存在，团队盘需确认读写权限。
  • 首次下载缓慢：属于GDrive正常现象，后续启动会直接读取缓存无需重复下载。

内容的提问来源于stack exchange，提问作者Mohammad Fasha