【ComfyUI】Pytorch预训练模型(torch.hub)缓存地址修改

最新推荐文章于 2024-06-28 11:48:44 发布
最新推荐文章于 2024-06-28 11:48:44 发布
阅读量1.7k 收藏 1
点赞数 3
分类专栏： stable diffusion AI绘画 comfyUI 文章标签： pytorch 缓存
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u013066244/article/details/133324932
版权

序言

最近玩ComfyUI时，每次生成图片，总是会下载一些东西，时间长了，C盘就不够用了，今天清理C盘发现，总是会在C:\Users\yutao\.cache\torch\hub\checkpoints这个路径下，下载大模型文件，关键还挺大的。

今天研究了下，怎么修改缓存路径，特记录下

修改方法

打开环境变量：

  1. 增加：XDG_CACHE_HOME=E:\cache

在这里插入图片描述
2. 增加：TORCH_HOME=%XDG_CACHE_HOME%\torch

在这里插入图片描述

再保存即可。

参考地址：

https://blog.csdn.net/JustPeanut/article/details/130165951

山鬼谣me
关注
  • 3
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
xception pytorch 预训练模型.zip
07-12
《Xception PyTorch 预训练模型详解与应用》 在深度学习领域，预训练模型已经成为提升模型性能和效率的重要工具。Xception是Google在2017年提出的一种深度卷积神经网络（CNN）架构，其设计灵感来源于Inception系列...
NeRF-pytorch预训练模型
11-07
**标题解析：** "NeRF-pytorch预训练模型" 指的是基于PyTorch实现的Neural Radiance Fields (NeRF)的预训练模型集合。NeRF是一种先进的3D场景表示方法，通过学习神经网络来捕捉和重建场景的几何形状和颜色信息。 **...
Pytorch预训练模型(torch.hub)缓存地址修改
JustPeanut的博客
04-15 4823
介绍了Pytorch是如何设置默认缓存路径的。
设置TORCH_HOME环境变量，改变下载和缓存的默认路径
llf000000的博客
04-24 451
【代码】设置TORCH_HOME环境变量，改变下载和缓存的默认路径。
如何清空Comfyui的gpu缓存
最新发布
向来缘浅，奈何情深
06-28 1650
本文介绍了在使用Comfyui绘图服务时，如何解决因显存未自动释放导致的爆显存问题。作者通过创建一个空载任务，在每次绘图任务完成后自动清理GPU显存，从而实现显存的自动释放。此外，文章还列举了其他清理显存的方法，如在启动命令后添加参数和使用ComfyUI-Easy-Use插件，但这些方法在作者的环境中未能奏效。
pytorch预训练模型下载保存路径和路径更改
q4326rgeqaadfd的博客
02-18 6680
pytorch 预训练模型下载
ComfyUI SDWebUI升级pytorch随记
applebomb的专栏
03-27 1192
目前使用的版本是去年10月的1.6版本，有点老。希望支持新的特性，于是乎开始作死。从升级torch开始
Pytorch中更改预训练权重文件的下载位置
孟孟单单的博客
05-16 6534
目录1. 参考链接2. 更改方法3. 一个小技巧 1. 参考链接 Pytorch更改预训练权重下载位置 pytorch———修改预训练模型下载路径 2. 更改方法 在线加载的预训练权重默认存放位置在：C:\Users\[用户名]\.cache\torch\hub\checkpoints 方法1：在某个.py文件最上方加上以下代码 import os # 更改预训练下载位置 os.environ['TORCH_HOME'] = '/data/Your_User_Name/pretrained_weig
手把手教你搭建更节省显存的AI绘画工具ComfyUI
白帽阿叁的博客
03-08 4708
感兴趣的小伙伴，赠送全套AIGC学习资料，包含AI绘画、AI人工智能等前沿科技教程和软件工具，具体看这里。AIGC技术的未来发展前景广阔，随着人工智能技术的不断发展，AIGC技术也将不断提高。未来，AIGC技术将在游戏和计算领域得到更广泛的应用，使游戏和计算系统具有更高效、更智能、更灵活的特性。同时，AIGC技术也将与人工智能技术紧密结合，在更多的领域得到广泛应用，对程序员来说影响至关重要。未来，AIGC技术将继续得到提高，同时也将与人工智能技术紧密结合，在更多的领域得到广泛应用。
PyTorch里面的torch.nn.Parameter()详解
09-18
在代码示例中提到的`self.v = torch.nn.Parameter(torch.FloatTensor(hidden_size))`，这里的`self.v`就被转换成了一个可训练的参数，它将作为模型的一部分参与训练过程。这意味着，在反向传播和优化过程中，`self.v...
浅谈pytorch torch.backends.cudnn设置作用
12-20
在PyTorch中，`torch.backends.cudnn`是一组用于控制CuDNN（NVIDIA CUDA Deep Neural Network库）行为的设置。CuDNN是一个针对深度神经网络（DNN）的高度优化库，它提供了大量的GPU加速功能，以提高深度学习模型的...
pytorch预训练模型vgg19-dcbb9e9d.txt
02-10
### PyTorch预训练模型VGG19-dcbb9e9d知识点解析 #### 一、PyTorch概述 PyTorch是一个基于Python的科学计算包，它为深度学习研究提供了灵活且强大的GPU加速。PyTorch由Facebook的人工智能研究实验室（FAIR）开发，...
pytorch预训练下载保存路径
weixin_42329982的博客
08-31 1594
pytorch下载的预训练参数保存路径： /.cache/torch/hub/checkpoints 或者 /.cache/torch/checkpoints
torch.hub.load 加载本地模型（已解决）
热门推荐
michael_quan2013的博客
03-29 1万+
不让 torch.hub.load 联网下载模型，改为 torch.hub.load 加载本地模型。本地模型默认的下载路径是：/root/.cache/torch/hub……运行网上的项目，有时会卡住或者超时，原因是 torch.hub.load 默认会去网上找模型，有时会因为网络问题而报错。
linux和windows的checkpoint保存地点
w18013886857的博客
10-22 805
linux和windows的checkpoint保存地点
Linux系统下pytorch预训练模型存放地址
qq_43345339的博客
08-24 397
cd ~/.cache/torch/hub/checkpoints/
【Stable Diffusion】ComfyUI整合包安装
白帽阿叁的博客
02-05 1403
感兴趣的小伙伴，赠送全套AIGC学习资料，包含AI绘画、AI人工智能等前沿科技教程和软件工具，具体看这里。AIGC技术的未来发展前景广阔，随着人工智能技术的不断发展，AIGC技术也将不断提高。未来，AIGC技术将在游戏和计算领域得到更广泛的应用，使游戏和计算系统具有更高效、更智能、更灵活的特性。同时，AIGC技术也将与人工智能技术紧密结合，在更多的领域得到广泛应用，对程序员来说影响至关重要。未来，AIGC技术将继续得到提高，同时也将与人工智能技术紧密结合，在更多的领域得到广泛应用。
ComfyUI中如何按照需求自动命名保存的图片/视频
聚梦小课堂
02-22 5109
在ComfyUI中，默认生成的图像都是堆在一起的，如果想要保存自己生成的图片为自己想要的名字，比如说每天生成的图片放到以当天日期命名的文件夹里边，或者用某个节点里边的某个变量来命名生成的文件或者文件夹，这些都并没有一个特别明显的设置。这篇文章就是教你如何按照自己的需要保存生成的图片、视频。
torch.hub.load 获取模型原理
05-31
`torch.hub.load` 是 PyTorch 提供的一个工具函数，用于从 GitHub 上的仓库中加载模型，并返回一个模型实例。其大致原理如下： 1. `torch.hub.load` 函数接受两个参数：`repo_or_dir` 和 `model_name`。`repo_or_dir` 可以是 GitHub 上的仓库地址，也可以是本地目录路径。如果是 GitHub 上的仓库地址，`torch.hub.load` 会通过 Git 下载仓库代码到本地。如果是本地目录路径，则直接加载该目录下的模型。 2. 加载模型需要使用模型的定义文件，通常是一个 Python 脚本或一个 Jupyter Notebook。`torch.hub.load` 会在仓库目录中寻找名为 `model_name.py` 或 `model_name.ipynb` 的文件，并执行该文件以获取模型定义。 3. 模型定义文件中通常包含一个 `load_model` 函数，用于加载训练好的模型参数，并返回一个模型实例。`torch.hub.load` 会调用该函数，并将其返回值作为模型实例返回给调用者。 总之，`torch.hub.load` 的作用是帮助用户方便地从 GitHub 上加载预训练的模型，并返回一个可用的模型实例。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

山鬼谣me

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值