问题标签 [libtorch]

我执行了下面的代码。但是 nonzero() 函数非常慢（需要 0.1 秒）。我的代码有什么问题？

libtorch 是否有任何方法可以获取张量或模型设备类型。

就像xxx.device在pytorch中一样？

我正在尝试使用 cmake 构建一个程序。由于几个原因，程序必须使用静态库而不是动态库构建，我需要使用 PyTorch，所以这就是我所做的：

1. 下载并安装 PyTorch 静态库（我libtorch.a在正确的路径中找到，在/home/me/pytorch/torch/lib）
2. 由CMakeLists.txt以下内容制成：

仅供参考，example-app.cpp是具有主要功能的文件，并且argparse/是一个目录，其中包含一些调用函数的源代码example-app.cpp

它一直工作到cmake -DCMAKE_PREFIX_PATH=/home/me/pytorch/torch ..，但以下build会引发一些错误，说它找不到对某些函数的引用，即以 . 开头的函数fbgemm::。fbgemm是（据我所知）用于实现 PyTorch 的某种 GEMM 库。

在我看来，在链接静态 PyTorch 库时，它的内部库之类的fbgemm东西没有正确链接，但我不是专家cmake，老实说也不完全确定。

我做错了什么，还是有解决这个问题的方法？任何帮助或朝着正确的方向推动将不胜感激。

附言

1. 确切的错误没有发布，因为它太长了，但它主要由undefined reference to ~错误组成。如果查看错误消息可能对某些人有帮助，我很乐意编辑问题并发布。

2. build如果我从代码中删除需要库函数的部分而不从.#include <torch/torch.h>example-app.cpp

我在 LibTorch 中有一个 CNN 结构，但尺寸不正确。我的目标是输入 3 通道 64x64 图像并为 DGAN 输出逻辑回归浮点数。最后一层我设置为输入通道 36，因为如果我移除该层，输出神经元的维度为 6x6，所以我猜这是全连接输入所需的维度。我想知道：

• 您通常会做什么来检查 LibTorch 或 Pytorch 中的尺寸（即检查最后一个模块所需的大小，检查每层有多少可训练参数......）
• 在这种情况下有什么错误

我得到的错误是：

我有一个包含一组图像的文件夹，我想在 LibTorch 中将它们作为 CustomDataset 加载。

自定义数据集.h

自定义数据集.cpp

即使当我在 CustomDataset 类中时，我看到成员变量m_filenames不为空，并且它get()提供了带有图像和标签的张量，但一旦我回到main.cpp名为 CustomDataset 的变量train_set的大小为 0。

我正在 LibTorch/Pytorch 中实现 DCGAN 网络。我正在关注https://github.com/pytorch/examples/blob/master/cpp/dcgan/dcgan.cpp中的官方示例。我的问题和示例之间的唯一区别是：

• 我的数据集由 RGB 图片（CelebA 数据集）组成，而示例中的图片是黑白图片（MNIST）
• 我的图片尺寸为 64x64，而 MNIST 图片为 28x28

这就是我的代码：

我不时保存小批量并绘制在生成器上输入噪声的结果。问题是，在 MNIST 示例中，结果是正确的，但在我的情况下，对于每张输出图片，我看到的是 9 张带有面孔而不是一张的较小图片（见附图）。

生成器怎么可能输出正确的形状但有 9 个几乎相同的面而不是一个？

我有一个张量数组，我想改变张量的形状。我尝试使用torch.view，但它引发了一个异常，即“shape[1] 对于大小为 10000 的输入无效”。任何人都可以给我有关错误信息的提示吗？

嗨，最近我已经安装了 Libtorch，并且可以在我的新 Visual Studio 项目中使用它而没有问题。目前我正在尝试在现有的 CUDA 项目中使用 Libtorch。但是当我包含火炬头时我遇到了这些奇怪的错误，我在互联网上找不到任何关于我的问题的解决方案。有谁知道这些错误的原因可能是什么？

环境

• 视窗 10
• CUDA 10.1
• 视觉工作室 2017
• C++14

此 PyTorch 模型评估在虚幻引擎中崩溃并出现错误 c0000374。

相同的代码在常规桌面应用程序上没有任何问题。模型是从插件的内容文件夹中加载的：

评估时出现错误：

输出窗口日志：

'UE4Editor.exe' (Win32)：已加载 'D:\projects\MyProject\Plugins\MyPlugin\Binaries\ThirdParty\MyPluginLibrary\Win64\torch_cpu.dll'。模块是在没有符号的情况下构建的。

“UE4Editor.exe”（Win32）：已加载“C:\Windows\System32\vcomp140.dll”。

检测到严重错误 c0000374

UE4Editor.exe 已触发断点。

尽我所能遵循此处的指南（C 驱动器不是 X，64 位），我已经达到了测试 Torch 的阶段

在尝试要求手电筒模块时，我收到此错误：

libtorch.dll 确实存在于 lua 所说的位置，并且错误消息没有给出任何无法加载它的原因。我现在已经重做了几次整个过程，同样的结果。谷歌搜索类似的问题，没有找到任何符合我的问题。完全不知道该做什么。

如果我使用luarocks install torch，我可以要求它，但是它无法识别我的 lapack 库并在torch.test(). 根据我正在关注的先前链接的教程，我应该通过下载torch rockspec文件，手动设置我的lapack库的路径，然后安装更改​​后的rockspec文件来解决这个问题，但是这样做会导致我正在写的这个问题.

我想做的就是制作意大利面条图像，但我无法让手电筒工作。我为含糊不清而道歉。