search

VITA-Group / FasterSeg

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
MIT License
525 stars 107 forks source link

Error about TensorRT and Pycuda #26

Closed NoOneUST closed 4 years ago

NoOneUST commented 4 years ago

When I run CUDA_VISIBLE_DEVICES=1 python train_search.py, I get this /export/data/lwangcg/FasterSeg/tools/utils/darts_utils.py:179: UserWarning: TensorRT (or pycuda) is not installed. compute_latency_ms_tensorrt() cannot be used. warnings.warn("TensorRT (or pycuda) is not installed. compute_latency_ms_tensorrt() cannot be used.")

I have tried to install Pycuda and TensorRT, but I might did not do it successfully. I get a warning in Pycuda test and fail in TensorRT test. Can you kindly help me with this? Note, I have no root privilege to the system(Centos 7)

FasterSeg: image

TensorRT: image

Pycuda: image

chenwydj commented 4 years ago

Hi @NoOneUST !

Please kindly refer to TensorRT doc and TensorRT github repo for installation issues. Thank you!