如何通过Torch Hub从私有Git仓库加载模型到Google Colab？

问题核心原因 #

torch.hub默认仅支持访问公开Git仓库，访问私有仓库时缺少权限验证，因此返回404错误。以下是两种可行的解决方法：

方法一：使用HTTPS个人访问令牌（PAT） #

1. 生成Git平台（如GitHub）的个人访问令牌：
   • 进入平台设置→开发者设置→个人访问令牌，创建新令牌，勾选repo权限，复制生成的令牌（仅显示一次，务必保存）
2. 在Colab中克隆私有仓库：

   !git clone https://<你的PAT>@github.com/EliasLeijon/myRepoName.git
   

3. 通过本地路径加载模型：

   import torch
   model = torch.hub.load('/content/myRepoName', 'bart.base', source='local')
   

方法二：使用SSH密钥（适合长期复用） #

1. 在Colab中生成SSH密钥对：

   !ssh-keygen -t rsa -b 4096 -C "你的邮箱地址"
   

   按提示连续回车即可，无需设置密码
2. 添加公钥到Git平台：
   • 查看公钥内容：!cat /root/.ssh/id_rsa.pub，复制输出的全部内容
   • 进入平台设置→SSH和GPG密钥，添加新的SSH密钥，粘贴公钥内容
3. 验证SSH连接：

   !ssh -T git@github.com
   

   出现Hi EliasLeijon! You've successfully authenticated...即表示验证成功
4. 克隆私有仓库并加载模型：

   !git clone git@github.com:EliasLeijon/myRepoName.git
   

   import torch
   model = torch.hub.load('/content/myRepoName', 'bart.base', source='local')
   

额外注意事项 #

• 确保你的私有仓库符合Torch Hub规范：根目录下必须存在hubconf.py文件，且该文件中定义了bart.base作为模型加载的入口函数
• 若直接使用torch.hub.load的git_url参数传入带PAT的链接，需注意令牌泄露风险，不建议在共享的Colab notebook中使用：

  import torch
  model = torch.hub.load(
      'EliasLeijon/myRepoName',
      'bart.base',
      git_url='https://<你的PAT>@github.com/EliasLeijon/myRepoName.git'
  )
  

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