修改Torch预训练模型下载目录

在使用PyTorch下载预训练模型时,其存储目录可以通过如下代码进行获取:

>>> import torch
>>> torch.hub.get_dir()

默认情况下Windows是在C盘的.cache目录下的。如果想修改其存储目录,可以通过环境变量TORCH_HOME进行修改。
修改环境变量后需要关闭终端再重新开一个新的终端,此时使用之前的代码会发现其目录修改为设置的目录了。