pytorch读取预训练模型并修改模型下载位置

1. 下载模型

import torchvision
model = torchvision.models.vgg13_bn(pretrained=True)
print(model)

2. 模型下载的位置默认在C盘，这个可以根据环境变量来设置。
   首先找到并打开 “查看高级系统设置” ，win10可以直接在下面的搜索导航中搜索到。
   
3. 打开之后进入环境变量
   
4. 点击‘新建’

5. 然后输入变量名 TORCH_HOME 和你所需要的储存位置，之后确定保存好。
   
6. 我使用的jupyter notebook, 所以需要重启jupyter notebook，pycharm,vscode等其他编辑器请自行尝试是否需要重启。