Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
6aa42e6e
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
403
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
6aa42e6e
编写于
4月 06, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #439 from Qsingle:fix_python36_pip_url
GitOrigin-RevId: 186ae708a144c92d4cec44411caf40d240e29a42
上级
39d98d45
f319d842
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
24 addition
and
10 deletion
+24
-10
cmake/cudnn.cmake
cmake/cudnn.cmake
+7
-4
cmake/tensorrt.cmake
cmake/tensorrt.cmake
+13
-6
scripts/whl/manylinux2014/init_image.sh
scripts/whl/manylinux2014/init_image.sh
+4
-0
未找到文件。
cmake/cudnn.cmake
浏览文件 @
6aa42e6e
...
...
@@ -6,7 +6,7 @@ endif()
if
(
"
${
CUDNN_ROOT_DIR
}
"
STREQUAL
""
AND NOT
"$ENV{CUDNN_ROOT_DIR}"
STREQUAL
""
)
set
(
CUDNN_ROOT_DIR $ENV{CUDNN_ROOT_DIR}
)
endif
()
message
(
"CUDNN ROOT: "
${
CUDNN_ROOT_DIR
}
)
if
(
MGE_CUDA_USE_STATIC AND NOT MGE_WITH_CUDNN_SHARED
)
find_library
(
CUDNN_LIBRARY
...
...
@@ -14,16 +14,19 @@ if(MGE_CUDA_USE_STATIC AND NOT MGE_WITH_CUDNN_SHARED)
PATHS
${
ALTER_LD_LIBRARY_PATHS
}
${
CUDNN_ROOT_DIR
}
${
PC_CUDNN_LIBRARY_DIRS
}
${
CMAKE_INSTALL_PREFIX
}
HINTS
${
ALTER_LIBRARY_PATHS
}
PATH_SUFFIXES lib lib64
PATH_SUFFIXES lib lib64
lib/x64
DOC
"CUDNN library."
)
else
()
if
(
MSVC
)
set
(
CMAKE_FIND_LIBRARY_SUFFIXES
".dll"
)
endif
()
find_library
(
CUDNN_LIBRARY
NAMES libcudnn.so libcudnn.dylib cudnn64
.dll
NAMES libcudnn.so libcudnn.dylib cudnn64
_8.dll cudnn.dll cudnn64_8 cudnn
PATHS
${
ALTER_LD_LIBRARY_PATHS
}
${
CUDNN_ROOT_DIR
}
${
PC_CUDNN_LIBRARY_DIRS
}
${
CMAKE_INSTALL_PREFIX
}
HINTS
${
ALTER_LIBRARY_PATHS
}
PATH_SUFFIXES lib lib64
PATH_SUFFIXES lib lib64
bin
DOC
"CUDNN library."
)
endif
()
...
...
cmake/tensorrt.cmake
浏览文件 @
6aa42e6e
...
...
@@ -18,22 +18,25 @@ if(MGE_CUDA_USE_STATIC)
PATH_SUFFIXES lib lib64
DOC
"TRT plugin library."
)
else
()
if
(
MSVC
)
set
(
CMAKE_FIND_LIBRARY_SUFFIXES
".dll"
)
endif
()
find_library
(
TRT_LIBRARY
NAMES libnvinfer.so libnvinfer.dylib nvinfer.dll
NAMES libnvinfer.so libnvinfer.dylib nvinfer.dll
nvinfer
PATHS
${
ALTER_LD_LIBRARY_PATHS
}
${
TRT_ROOT_DIR
}
${
CMAKE_INSTALL_PREFIX
}
HINTS
${
ALTER_LIBRARY_PATHS
}
PATH_SUFFIXES lib lib64
DOC
"TRT library."
)
find_library
(
TRT_PLUGIN_LIBRARY
NAMES libnvinfer_plugin.so libnvinfer_plugin.dylib nvinfer_plugin.dll
NAMES libnvinfer_plugin.so libnvinfer_plugin.dylib nvinfer_plugin.dll
nvinfer_plugin
PATHS
${
ALTER_LD_LIBRARY_PATHS
}
${
TRT_ROOT_DIR
}
${
CMAKE_INSTALL_PREFIX
}
HINTS
${
ALTER_LIBRARY_PATHS
}
PATH_SUFFIXES lib lib64
DOC
"TRT plugin library."
)
endif
()
message
(
"TRT_LIBRARY"
${
TRT_LIBRARY
}
)
if
(
TRT_LIBRARY STREQUAL
"TRT_LIBRARY-NOTFOUND"
)
message
(
FATAL_ERROR
...
...
@@ -121,7 +124,8 @@ if(TensorRT_VERSION_MAJOR GREATER_EQUAL 7)
if
(
MGE_CUDA_USE_STATIC
)
find_library
(
LIBMYELIN_COMPILER
NAMES libmyelin_compiler_static.a myelin_compiler_static.lib
NAMES libmyelin_compiler_static.a myelin_compiler_static.lib myelin64_1.lib
myelin64_1
PATHS
${
__found_trt_root
}
/lib
)
if
(
LIBMYELIN_COMPILER STREQUAL
"LIBMYELIN_COMPILER-NOTFOUND"
)
message
(
FATAL_ERROR
"Can not find LIBMYELIN_COMPILER Library"
)
...
...
@@ -134,7 +138,8 @@ if(TensorRT_VERSION_MAJOR GREATER_EQUAL 7)
find_library
(
LIBMYELIN_EXECUTOR
NAMES libmyelin_executor_static.a myelin_executor_static.lib
NAMES libmyelin_executor_static.a myelin_executor_static.lib myelin64_1.lib
myelin64_1
PATHS
${
__found_trt_root
}
/lib
)
if
(
LIBMYELIN_EXECUTOR STREQUAL
"LIBMYELIN_EXECUTOR-NOTFOUND"
)
message
(
FATAL_ERROR
"Can not find LIBMYELIN_EXECUTOR Library"
)
...
...
@@ -148,6 +153,7 @@ if(TensorRT_VERSION_MAJOR GREATER_EQUAL 7)
find_library
(
LIBMYELIN_PATTERN_RUNTIME
NAMES libmyelin_pattern_runtime_static.a myelin_pattern_runtime_static.lib
myelin64_1.lib myelin64_1
PATHS
${
__found_trt_root
}
/lib
)
if
(
LIBMYELIN_PATTERN_RUNTIME STREQUAL
"LIBMYELIN_PATTERN_RUNTIME-NOTFOUND"
)
message
(
FATAL_ERROR
"Can not find LIBMYELIN_PATTERN_RUNTIME Library"
)
...
...
@@ -162,6 +168,7 @@ if(TensorRT_VERSION_MAJOR GREATER_EQUAL 7)
find_library
(
LIBMYELIN_PATTERN_LIBRARY
NAMES libmyelin_pattern_library_static.a myelin_pattern_library_static.lib
myelin64_1.lib myelin64_1
PATHS
${
__found_trt_root
}
/lib
)
if
(
LIBMYELIN_PATTERN_LIBRARY STREQUAL
"LIBMYELIN_PATTERN_LIBRARY-NOTFOUND"
)
message
(
FATAL_ERROR
"Can not find LIBMYELIN_PATTERN_LIBRARY Library"
)
...
...
@@ -175,7 +182,7 @@ if(TensorRT_VERSION_MAJOR GREATER_EQUAL 7)
else
()
find_library
(
LIBMYELIN_SHARED
NAMES libmyelin.so myelin.dll
NAMES libmyelin.so myelin.dll
myelin64_1.dll myelin64_1
PATHS
${
__found_trt_root
}
/lib
)
if
(
LIBMYELIN_SHARED STREQUAL
"LIBMYELIN_SHARED-NOTFOUND"
)
...
...
scripts/whl/manylinux2014/init_image.sh
浏览文件 @
6aa42e6e
...
...
@@ -2,6 +2,7 @@
GET_PIP_URL
=
'https://bootstrap.pypa.io/get-pip.py'
GET_PIP_URL_35
=
'https://bootstrap.pypa.io/pip/3.5/get-pip.py'
GET_PIP_URL_36
=
'https://bootstrap.pypa.io/pip/3.6/get-pip.py'
SWIG_URL
=
'https://codeload.github.com/swig/swig/tar.gz/refs/tags/rel-3.0.12'
LLVM_URL
=
'https://github.com/llvm-mirror/llvm/archive/release_60.tar.gz'
CLANG_URL
=
'https://github.com/llvm-mirror/clang/archive/release_60.tar.gz'
...
...
@@ -29,6 +30,9 @@ do
PIP_URL
=
${
GET_PIP_URL
}
if
[
${
ver
}
=
"35m"
]
;
then
PIP_URL
=
${
GET_PIP_URL_35
}
else if
[
${
ver
}
=
"36m"
]
;
then
PIP_URL
=
${
GET_PIP_URL_36
}
fi
fi
echo
"use pip url:
${
PIP_URL
}
"
curl
${
PIP_URL
}
| /opt/python/cp
${
python_ver
}
-cp
${
ver
}
/bin/python -
\
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录