fix import issue on no-gpu machine