未验证 提交 51c66a67 编写于 作者: L Leo Chen 提交者: GitHub

add path of libcuda.so.1 for cinn compilation (#50268)

上级 caf2008b
...@@ -165,6 +165,8 @@ function cmake_base() { ...@@ -165,6 +165,8 @@ function cmake_base() {
-DPYTHON_LIBRARIES:FILEPATH=/opt/conda/lib/libpython3.so" -DPYTHON_LIBRARIES:FILEPATH=/opt/conda/lib/libpython3.so"
/opt/conda/bin/pip install -r ${PADDLE_ROOT}/python/requirements.txt /opt/conda/bin/pip install -r ${PADDLE_ROOT}/python/requirements.txt
fi fi
# for CINN, to find libcuda.so.1
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-11.2/compat/
else else
pip install -r ${PADDLE_ROOT}/python/requirements.txt pip install -r ${PADDLE_ROOT}/python/requirements.txt
fi fi
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册