search

Cw-zero / TensorRT_yolo3_module

You can import this module directly
55 stars 19 forks source link

Why pytorch? #10

Open zhzhang747 opened 4 years ago

zhzhang747 commented 4 years ago

In file trt_yolo3_module_1batch.py: Ln 49 a = torch.cuda.FloatTensor() #pytorch必须首先占用部分CUDA Why need pytorch first? Thank you.

Scheaven commented 4 years ago

你解决了吗？我到这一行报错： TypeError: type torch.cuda.FloatTensor not available. Torch not compiled with CUDA enabled.