在屏幕上的窗口中显示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个维度,特别是(宽度、高度、通道)的顺序。 此布局限制将在将来的版本中放宽。在
- 项目
标签: