在屏幕上的窗口中显示PyTorch图像张量
TorchWindow的Python项目详细描述
火炬窗
TorchWindow是一个主要用C编写的pythorchwindow库，可以通过Vulkan Cuda interop直接从GPU内存（GPU和CPU之间没有来回复制）查看屏幕上的PyTorch Cuda张量。在
构建需求
目前TorchWindow只针对64位平台。将来可能会增加32位。在
Windows
Visual Studio
CUDA SDK
Vulkan SDK
CMake公司
Linux
Linux目前不受支持，但将在下一个版本中提供。在
MacOS
MacOS不受支持，但将来可能会支持。在
从源安装
在项目根目录中：
mkdir build
cd build
cmake ..
cmake --build .
cd ..
pip install .
从二进制文件安装
^{pr2}$使用
要创建窗口：
from torchwindow import Window
window = Window(640, 480, name="Torch Window")
在窗口中显示图像张量
window.draw(image_tensor)
请注意，图像张量必须有3个维度，特别是（宽度、高度、通道）的顺序。 此布局限制将在将来的版本中放宽。在
- 项目
标签：
欢迎加入QQ群-->： 979659372
