torchvison下载模型并更改存储路径

已于 2022-07-06 19:59:45 修改
阅读量2k 收藏 10
点赞数 9
分类专栏： pytorch学习 文章标签： python pycharm
于 2022-07-06 19:56:34 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_43571321/article/details/125646524
版权
本文介绍了使用PyCharm时，torchvision下载模型更改存储路径的方法。因原存储在C盘，为减轻其压力，提供两种方法：一是在环境变量中创建TORCH_HOME；二是直接在PyCharm的torch源码中更改路径，还给出了让模型存于文件夹的设置方式。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

torchvision下载模型并更改存储的路径（本来是存储在c盘）

  • 使用的Ide是pycharm

  • 如果想要知道模型的存储路径可以pycharm控制台上输入，查看自己的路径

import torch
torch.hub.get_dir()

更改存储路径目的是想让C盘减少压力，所以下面提供两种方法：

  • 第一种，在环境变量中创建TORCH_HOME=”YourPath“ (第一种方法我用了但是没效果)

    • 打开系统属性
      在这里插入图片描述

    • 打开环境变量
      在这里插入图片描述

    • 新建系统变量
      在这里插入图片描述

    • 填写确定并返回，重启PyCharm

    在这里插入图片描述

  • 第二种是直接在PyCharm的torch源码中更改路径，以VGG16模型为例

    • 点进去源码

在这里插入图片描述

  • 可以看到这里只有一个return语句

在这里插入图片描述

  • 这里中的第2点应该是存储路径的函数

在这里插入图片描述

  • 看到这个函数model_dir这个变量是空的，所以会获取目录

在这里插入图片描述
在这里插入图片描述

  • 我们可以把model_dir改为自己的路径
    在这里插入图片描述

  • 效果是这样，可以看到是没有文件夹的
    在这里插入图片描述

  • 那如果让模型放在一个文件夹里可以这样，首先model_dir改回None，然后hub_dir改成自己的路径，保存就可以了。
    在这里插入图片描述

在这里插入图片描述

  • 最后下载的时候就是这样

    在这里插入图片描述

pytorch 模型下载，郑重推荐
jacke121的专栏
08-06 5208
    http://download.pytorch.org/models/ http://download.pytorch.org/models/densenet121-a639ec97.pth   下载安装pytorch： pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-...
[Pytorch系列-42]：工具集 - torchvision常见预训练模型的下载地址
热门推荐
文火冰糖（王文兵）的博客
11-06 1万+
作者主页(文火冰糖的硅基工坊)：文火冰糖（王文兵）的博客_文火冰糖的硅基工坊_CSDN博客 本文网址： 目录 步骤1：torchvision概述 步骤2：如何获取框架提供的预训练模型 步骤3：通过IE浏览器手工下载模型 步骤4：加载模型 步骤1：torchvision概述 [Pytorch系列-37]：工具集 - torchvision库详解(数据集、数据预处理、模型)_文火冰糖（王文兵）的博客-CSDN博客作者主页(文火冰糖的硅基工坊)：文火冰糖（王文兵）的博客_文火冰糖的硅基工坊_CSD
pytorch 清楚缓存_Pytorch快速下载预训练模型并修改保存路径
weixin_39761696的博客
01-14 1212
【Pytorch】快速下载预训练模型并修改保存路径首次用Pytorch加载预训练模型，需要在线下载，但是下载速度比较慢。下载后会保存在本地缓存里。如果能直接加载本地下载好的模型就会快了，主要是个修改路径的问题。所以要提升速度一般有两种方法：1.修改torch源码，一次性改变下载url2.将离线模型权重存到缓存文件夹里参考：pytorch预训练模型的下载地址以及解决下载速度慢的方法 - you-wh...
pytorch预训练模型下载保存路径更改
最新发布
岁月如歌
02-18 308
如果没有预先下载好预训练模型，在运行这个代码后，自动下载预训练模型的。
更改torchvision预训练模型保存路径
weixin_45406060的博客
10-21 630
更改torchvision已训练模型保存路径
pytorch torchvision中预训练模型下载地址
x_yan033的博客
03-11 1432
Resnet: model_urls = { 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https...
torchvision数据集下载位置修改
qq_44199379的博客
10-20 1270
torchvision常用数据集加载介绍及数据集下载位置修改
【Pytorch】快速下载预训练模型并修改保存路径
qq_17047265的博客
05-05 6634
首次用Pytorch加载预训练模型，需要在线下载，但是下载速度比较慢。下载后会保存在本地缓存里。如果能直接加载本地下载好的模型就会快了，主要是各修改路径的问题。 所以要提升速度一般有两种方法： 1.修改torch源码，一次性改变下载url 2.将离线模型权重存到缓存文件夹里 参考：pytorch预训练模型的下载地址以及解决下载速度慢的方法 - you-wh - 博客园 但是用云服务器时候这两种...
Pytorch（二）
hello_world_juns的博客
11-07 1305
DataLoadr是Pytorch中数据读取的一个重要的接口，该接口的目的：将自定义的Dataset根据batch_size大小，是否shuffle等封装成一个Batch_Size大小的Tensor，用于后面的训练。DataLoader(object)的参数dataset(Dataset): 传入的数据集batch_size(int, optional): 每个batch有多少个样本shuffle(bool, optional): 在每个epoch开始的时候，对数据进行重新排序。
pytorch 从放弃到入门（四）
xpwmiracle的博客
05-25 666
没头脑和不高兴
Pytorch入门(B站小土堆）
m0_53956610的博客
11-08 637
read_data.py定义了一个自定义的 PyTorch 数据集类MyData，用于加载图像数据。import osDataset是 PyTorch 数据集类的基类，用于创建自定义数据集。Image来自 Pillow 库，用于图像处理。os用于文件和目录操作__init__方法初始化数据集对象，接收两个参数：root_dir: 数据集根目录。label_dir: 类别目录，存储了具体的图像。self.path是类别目录的完整路径。存储了类别目录中所有图像的文件名列表。
pytorch学习笔记(Tensor,Variable,nn.Module,optim)
eilot_c的博客
01-30 3890
Tensor(张量) torch.Tensor 默认的是 torch.FloatTensor 数据类型,也可以定义我们自己想要的数据类型 当然也可以创建一个全是0的空的Tensor或者取一个正态分布作为随机初始值 可以像numpy一样通过索引的方式取得其中的元素,同时改变它的值 除此之外可以在Tensor和numpy.ndarray之间相互转换: 判断是否支持GPU,如果...
torch、torchvision离线下载网址
gengmeng_的博客
11-24 1746
https://download.pytorch.org/whl/torch/ https://download.pytorch.org/whl/torchvision/
基于Anaconda创建的虚拟环境，修改预训练模型下载的路径
weixin_44984048的博客
01-23 1455
pytorch修改预训练模型的保存路径
pytorch预训练模型下载保存路径和路径更改
q4326rgeqaadfd的博客
02-18 8459
pytorch 预训练模型下载
修改模型参数的下载地址
qq_44814543的博客
10-30 693
修改模型参数的下载地址
torch torchvision 下载地址
03-07 1083
然后找：Compatible tags: 对应的是CPXXX?比如对应的是 CP38 ,那么就去下载 CP38的链接。
解决torchvision.model下载预训练模型太慢的问题
major_in_data_的博客
01-16 945
model和.pth如果是对应的，就可以用model.load_state_dict加载。注意这条语句是在模型上直接修改，不应写成model = model.load_state_dict。所以我们可以自己在浏览器下载模型，然后加载进去。Ctrl+函数打开源码自己就可以找到。当pretrained=True，才会联网下载模型，否则很快，仅得到一个没训练过的模型。.pth文件或者state_dict变量：模型参数，里面是模型每一层具体的浮点数。model：模型，不含参数。
Pytorch加载torchvision从本地下载好的预训练模型的简单解决方案
weixin_43178406的博客
11-21 1万+
本文主要介绍了Pytorch加载torchvision从本地下载好的预训练模型的简单解决方案，希望能对新手有所帮助。之所以从本地加载下载好的模型，是因为默认是从Pytorch官网进行下载，但是它并不支持断点续传，而且单线程下载速度很慢，所以要么出现RemoteDisconnected如下图所示，要么出现半途下载中断的情况。但遗憾的是网上已有的解决方案是让修改源码，但这样非常不优雅，而且对新手不友好。个人找到了比较简单而且有效的解决方案，特意分享给大家，希望能对大家有所帮助。 1. 问题描述 2. 解决方案
torchvison conda下载
01-16
此命令同样来源于 PyTorch 官方渠道，用于获取并安装最新的 `torchvision` 版本[^2]。 对于特定 CUDA 版本的需求场景下，比如需要兼容 CUDA 9.0 的环境，则可以通过指定 `cudatoolkit` 参数来完成安装操作： ```...
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值