MegEngine / MegCC

MegCC是一个运行时超轻量,高效,移植简单的深度学习模型编译器
Apache License 2.0
474 stars 56 forks source link

prepare.sh 编译 nvcc #28

Closed tpoisonooo closed 1 year ago

tpoisonooo commented 1 year ago

我的环境里有 TRT_HOME / CUDA_HOME/ CUDNN_HOME 这些变量,似乎 prepare.sh 期间开始用 nvcc build 东西。

看 megcc README 应该不涉及 nvidia 的东西。

是否应该把 megbrain build 那里裁一下?

附我的 env

CUDA_ROOT=/usr/local/cuda
CUDA_HOME=/usr/local/cuda
GIT_EDITOR=vim
TRT_HOME=/home/khj/trt8431/TensorRT-8.4.3.1
CPLUS_INCLUDE_PATH=:/home/khj/trt8431/TensorRT-8.4.3.1/include:/usr/local/cuda/include:/home/khj/cudnn84/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/include
LD_LIBRARY_PATH=:/home/khj/trt8431/TensorRT-8.4.3.1/lib:/home/khj/cudnn84/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive/lib
CUDNN_HOME=/home/khj/cudnn84/cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive
DOCKER_HOST=unix:///run/user/1000/docker.sock
CONDA_EXE=/home/khj/miniconda3/bin/conda
_CE_M=
_CE_CONDA=
CONDA_PYTHON_EXE=/home/khj/miniconda3/bin/python
CONDA_SHLVL=1
CONDA_PREFIX=/home/khj/miniconda3
CONDA_DEFAULT_ENV=base
CONDA_PROMPT_MODIFIER=(base)
chenqy4933 commented 1 year ago

按道理说应该不会执行CUDA相关编译了,因为 https://github.com/MegEngine/MegCC/blob/main/third_party/prepare.sh#L46 这里已经显式的指定 CUDA 已经关了,会不会是你系统中其他进程在使用 nvcc ?

tpoisonooo commented 1 year ago

遇到了找不到 cuda header 的报错。

tpoisonooo commented 1 year ago

cuda_runtime_api.h 设置好之后,现在是对的。

chenqy4933 commented 1 year ago

看来 MegEngine 的 CMakeList 可能有点问题, cc @haolongzhangm

tpoisonooo commented 1 year ago

BTW, https://github.com/MegEngine/MegCC/pull/27 这个俺没要改的了。