libtorch cuda is not available

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 xiaoxifuhongse opened 3 years ago

xiaoxifuhongse commented 3 years ago

I have cuda10.2, and download the libtorch1.7.1, torch::cuda::is_available() return 0 ,can you help me?