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

  1. 下载模型
import torchvision
model = torchvision.models.vgg13_bn(pretrained=True)
print(model)
  1. 模型下载的位置默认在C盘,这个可以根据环境变量来设置。
    首先找到并打开 “查看高级系统设置” ,win10可以直接在下面的搜索导航中搜索到。
    pytorch读取预训练模型并修改模型下载位置_第1张图片
  2. 打开之后进入环境变量
    pytorch读取预训练模型并修改模型下载位置_第2张图片
  3. 点击‘新建’

pytorch读取预训练模型并修改模型下载位置_第3张图片

  1. 然后输入变量名 TORCH_HOME 和你所需要的储存位置,之后确定保存好。
    pytorch读取预训练模型并修改模型下载位置_第4张图片
  2. 我使用的jupyter notebook, 所以需要重启jupyter notebook,pycharm,vscode等其他编辑器请自行尝试是否需要重启。

你可能感兴趣的:(pytorch,pytorch,python)