Add conda package (#231)
* Make sure to install the nnvm_compiler library. * Add conda packge.
This commit is contained in:
Родитель
c6c2e15663
Коммит
a0b0546a29
|
@ -113,8 +113,11 @@ if(BUILD_SHARED_NNVM)
|
|||
if(WIN32)
|
||||
install(TARGETS nnvm RUNTIME DESTINATION bin)
|
||||
install(TARGETS nnvm ARCHIVE DESTINATION lib)
|
||||
install(TARGETS nnvm_compiler RUNTIME DESTINATION bin)
|
||||
install(TARGETS nnvm_compiler ARCHIVE DESTINATION lib)
|
||||
else()
|
||||
install(TARGETS nnvm LIBRARY DESTINATION lib)
|
||||
install(TARGETS nnvm_compiler LIBRARY DESTINATION lib)
|
||||
endif()
|
||||
endif()
|
||||
if(BUILD_STATIC_NNVM)
|
||||
|
|
|
@ -0,0 +1,28 @@
|
|||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
|
||||
if [ -z "$PREFIX" ]; then
|
||||
PREFIX="$CONDA_PREFIX"
|
||||
fi
|
||||
|
||||
if [ "$(uname)" = 'Darwin' ]
|
||||
then
|
||||
# Without this, Apple's default shipped clang will refuse to see any
|
||||
# headers like mutex.
|
||||
export MACOSX_DEPLOYMENT_TARGET=10.9
|
||||
fi
|
||||
|
||||
rm -rf build || true
|
||||
mkdir -p build
|
||||
cd build
|
||||
# Enable static-libstdc++ to make it easier to link this library with
|
||||
# other C++ compilers
|
||||
CXXFLAGS=-static-libstdc++ cmake -DCMAKE_PREFIX_PATH=${PREFIX} -DCMAKE_INSTALL_PREFIX=${PREFIX} ..
|
||||
make -j4 VERBOSE=1
|
||||
make install/fast
|
||||
cd ..
|
||||
|
||||
cd python
|
||||
$PYTHON setup.py install
|
||||
cd ..
|
|
@ -0,0 +1,32 @@
|
|||
{% set version = "0.1.dev" %}
|
||||
|
||||
package:
|
||||
name: nnvm
|
||||
version: {{ version }}
|
||||
|
||||
source:
|
||||
path: ../..
|
||||
|
||||
build:
|
||||
number: 1
|
||||
skip: True # [win]
|
||||
|
||||
requirements:
|
||||
build:
|
||||
- cmake
|
||||
- python >=3
|
||||
- numpy
|
||||
- setuptools
|
||||
- nose
|
||||
- decorator
|
||||
run:
|
||||
- tvm
|
||||
- topi
|
||||
- python >=3
|
||||
- numpy
|
||||
- decorator
|
||||
|
||||
about:
|
||||
home: https://github.com/dmlc/nnvm
|
||||
license: Apache2
|
||||
summary: Bring deep learning to bare metal
|
Загрузка…
Ссылка в новой задаче