pytorch 相关问题

CUDA、PyTorch 和 YOLO 集成问题

我安装了Python 3.11，然后用CUDA重新编译了OpenCV。我也在使用 CUDA 开发 YOLO，但现在当我使用 CUDA 运行 PyTorch 12.4 时，我遇到了 YOLO 错误： 无法加载系统...

有条件地更改 RGB 图像中的像素值，然后删除通道

我只是想比较图像的像素，如果该像素是粉红色的（R值= 0.502，G值= 0.0，B值= 0.502），则将其更改为黑色，否则将其更改为白色。之后...

如何在 PyTorchLightning 中手动指定检查点路径

目前我正在使用 TensorBoardLogger 来满足我的所有需求，它很完美，但我不喜欢它处理检查点命名的方式。我希望能够指定文件名和文件夹...

在 Pytorch 中通过字典映射张量元素

我有一个张量是： t = torch.tensor([1, 2, 3]) 和映射： 映射 = {1: 0.2, 2: 1.2, 3: 3.0} 我想通过映射来映射 t 中的元素，因此预期结果将是 火炬.张量（[0.2，...

在 pytorch 中，如何并行化（在 GPU 上）重复执行的一组布尔函数？

我有一组独立的布尔函数，并且（假设）可以并行执行。我想重复调用这些相同的函数。请参阅下面的代码，其中的输出...

RTX 3080 出现分段错误（核心转储）

嗨~我刚拿到RTX 3080，遇到了很多版本问题。其中之一是弹出“分段错误（核心转储）”的问题和多处理问题。 下面是我的设备列表： 显卡：...

将布尔掩码与 PyTorch 张量中的切片相结合

我想用布尔掩码和普通索引来索引 pytorch 张量。像这样的东西： 我 = 2 j = 0 掩码 = torch.randn(480, 360, 3) > 0 张量 = torch.zeros(480, 360, 4, 80) 十...

如何在 PyTorch 中结合切片分配、掩码分配和广播？

更具体地说，我想知道如何在 PyTorch 中同时按切片和掩码在不同维度上分配张量。 这是一个关于我想做的事情的小例子： 有了张量...

为什么我的DataLoader进程使用了高达2.6GB的虚拟内存，有什么方法可以减少它吗？

为什么我的DataLoader进程使用了高达2.6GB的虚拟内存，有什么方法可以减少它吗？ 每个DataLoader进程占用2.6GB虚拟内存，4个进程占用10.4GB。 来自

为嵌套在多个文件夹中的数据创建训练、测试拆分

我正在准备数据来训练图像识别模型。我目前有一个文件夹（数据集），其中包含多个带有标签名称的文件夹，这些文件夹包含图像

将 pytorch 数据加载器加载到 GPU 中

有没有办法将 pytorch DataLoader (torch.utils.data.Dataloader) 完全加载到我的 GPU 中？ 现在，我将每个批次分别加载到我的 GPU 中。 CTX = torch.device('cuda') train_loader = torch.util...

如何为 PyTorch TensorBoard 将多条 PR 曲线绘制到一张图表上？

下面是绘制PR曲线的代码取自https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#assessing-trained-models-with-tensorboard 从 torch.utils.tensorboard 导入

没有名为“torch._six”的模块可导入队列

我正在尝试应用来自 github https://github.com/sanghyun-son/EDSR-PyTorch 的 RCAN 模型。我正在使用 pytorch 2.0.1。我收到错误： ModuleNotFoundError：没有名为“torch._six”的模块 ...

无法训练我的 UNET 多类分割模型

我尝试用 pytorch 从头开始制作我的 UNET，但什么也没得到，但黑色蒙版作为我的模型的输出。我需要对汽车的损坏进行分段，因此我实现了彩色贴图。我确信 70% 的事情是......

如何修复调用 `cublasSgemm` 时的 RuntimeError CUDA 错误 CUBLAS_STATUS_INVALID_VALUE？

在工作的 cuda 环境上训练某些模型时，调用 cublasSgemm(handle, opa, opb, m, n, k, &alpha,... 时会出现错误 RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE

如何可视化带边缘的分段任意模型？

我可以使用分段任意模型并可视化其结果，但它们看起来与在线演示不同。 我的结果看起来像这样： 这是该部分的结果 -

Pytorch + Ray Tune 报告 ImplicitFunc 太大，不知道哪个引用大

与这个问题类似，Ray Tune 正在向我报告： ValueError：参与者 ImplicitFunc 太大 (421 MiB > FUNCTION_SIZE_ERROR_THRESHOLD=95 MiB)。检查它的定义是否隐式...

参数和子项有什么区别？

看起来parameters和children显示的信息是一样的，那么它们之间有什么区别呢？ 进口火炬 print('火炬.__版本__', 火炬.__版本__) m = torch.load('imagenet_resnet18.pth...

计算滑动“图像相似度”分数，类似于 conv2d

我有一个大图像和一个较小的“内核”图像。我想将内核与图像的每个部分进行比较（通过在图像上“滑动”内核）并检索“

我尝试使用pytorchchildren()将resnet分成两部分，但它不起作用

这是一个简单的例子。我尝试使用子节点将网络（Resnet50）分为两部分：头部和尾部。从概念上讲，这应该可行，但事实并非如此。这是为什么呢？ 进口火炬 导入 torch.nn...