当执行下面这条语句的时候

vgg16_true=torchvision.models.vgg16(pretrained=True)

会默认启动该模型的安装,我发现它的默认路径是在C盘

如何更改默认路径?

方式一:修改环境变量(永久生效)

  1. 在Windows搜索栏输入“高级系统设置”,点击进入“系统属性”窗口。

  2. 在“高级”选项卡下,点击“环境变量”。

  3. 在“系统变量”(对所有用户生效)或“用户变量”(仅对当前用户生效)区域,点击“新建”。

  4. 在弹出的窗口中:

    • 变量名:输入 TORCH_HOME

    • 变量值:输入你想设置的新路径。例如 E:\PyTorchModels

  5. 点击“确定”保存所有设置。

方式二:当前会话中生效

可以在加载模型前,用 os.environ 设置 TORCH_HOME 变量

# 在加载模型前,先设置下载路径
os.environ['TORCH_HOME'] = 'E:/MyProject/PyTorchModels'

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐