search

dangnh0611 / kaggle_rsna_breast_cancer

1st place solution of RSNA Screening Mammography Breast Cancer Detection competition on Kaggle: https://www.kaggle.com/competitions/rsna-breast-cancer-detection
MIT License
75 stars 25 forks source link

Building docker image #4

Closed ramon349 closed 12 months ago

ramon349 commented 1 year ago

Hello,

Thank you for sharing your repository and details on running your code. I've been having issues building the docker image where the final Torch2trt fails due to missing the CUDA_HOME variable. I've tried the install process without Torch2trt but found that Cuda is unavailable. Do you have any suggestions on how to remedy this issue?

Best,

dangnh0611 commented 1 year ago

Hi, was CUDA available on your system, e.g nvcc -V You can skip Torch2Trt installation if you want, and also no TensorRT at all. But you need to modify some code to make it work with Torch model instead of TensorRT, especially the YOLOX related-parts.