如何自定义pytorch中的模型下载路径
·
当执行下面这条语句的时候
vgg16_true=torchvision.models.vgg16(pretrained=True)
会默认启动该模型的安装,我发现它的默认路径是在C盘
如何更改默认路径?
方式一:修改环境变量(永久生效)
-
在Windows搜索栏输入“高级系统设置”,点击进入“系统属性”窗口。
-
在“高级”选项卡下,点击“环境变量”。
-
在“系统变量”(对所有用户生效)或“用户变量”(仅对当前用户生效)区域,点击“新建”。
-
在弹出的窗口中:
-
变量名:输入
TORCH_HOME。 -
变量值:输入你想设置的新路径。例如 E
:\PyTorchModels。
-
-
点击“确定”保存所有设置。
方式二:当前会话中生效
可以在加载模型前,用 os.environ 设置 TORCH_HOME 变量
# 在加载模型前,先设置下载路径
os.environ['TORCH_HOME'] = 'E:/MyProject/PyTorchModels'
更多推荐




所有评论(0)