torchvison下载模型并更改存储路径

torchvision下载模型并更改存储的路径（本来是存储在c盘）

• 使用的Ide是pycharm

• 如果想要知道模型的存储路径可以pycharm控制台上输入，查看自己的路径

import torch
torch.hub.get_dir()

更改存储路径目的是想让C盘减少压力，所以下面提供两种方法：

• 第一种，在环境变量中创建TORCH_HOME=”YourPath“ (第一种方法我用了但是没效果)

  • 打开系统属性
    

  • 打开环境变量
    

  • 新建系统变量
    

  • 填写确定并返回，重启PyCharm

  

• 第二种是直接在PyCharm的torch源码中更改路径，以VGG16模型为例

  • 点进去源码

• 可以看到这里只有一个return语句

• 这里中的第2点应该是存储路径的函数

• 看到这个函数model_dir这个变量是空的，所以会获取目录

• 我们可以把model_dir改为自己的路径
  

• 效果是这样，可以看到是没有文件夹的
  

• 那如果让模型放在一个文件夹里可以这样，首先model_dir改回None，然后hub_dir改成自己的路径，保存就可以了。
  

• 最后下载的时候就是这样