wsx000 / YOLOv4-LibTorch-GPU-deployment

This is a YOLOv4 post-processing code used gpu based on pure libtorch API

GNU General Public License v2.0

14 stars 7 forks source link

Open hello968 opened 3 years ago

hello968 commented 3 years ago

include "torchvision/cpu/vision_cpu.h"

这是自己写的头文件吗？我找不到这个文件

wsx000 commented 3 years ago

这是torchvision自带的头文件，libtorch里没有的，你需要自行编译安装torchvision

mengyao-wang commented 2 years ago

这是torchvision自带的头文件，libtorch里没有的，你需要自行编译安装torchvision

请问，为什么我安装torchvision之后没有cpu或cuda文件夹，我的pytorch是1.2.0, libtorch是1.3.0, vision是0.4.1

wsx000 commented 2 years ago

这是torchvision自带的头文件，libtorch里没有的，你需要自行编译安装torchvision

请问，为什么我安装torchvision之后没有cpu或cuda文件夹，我的pytorch是1.2.0, libtorch是1.3.0, vision是0.4.1

我感觉可能你的pytorch版本跟vision的版本不对应