diff --git a/.bazelrc b/.bazelrc new file mode 100644 index 0000000..f974751 --- /dev/null +++ b/.bazelrc @@ -0,0 +1,12 @@ +build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true +build:manylinux2010cuda100 --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.0:toolchain +build:manylinux2010cuda101 --crosstool_top=//third_party/toolchains/preconfig/ubuntu16.04/gcc7_manylinux2010-nvcc-cuda10.1:toolchain +build --spawn_strategy=standalone +build --strategy=Genrule=standalone +build -c opt +build --action_env TF_HEADER_DIR="/usr/local/google/home/jaeyoo/workspace/cpu_venv/lib/python3.8/site-packages/tensorflow/include" +build --action_env TF_SHARED_LIBRARY_DIR="/usr/local/google/home/jaeyoo/workspace/cpu_venv/lib/python3.8/site-packages/tensorflow" +build --action_env TF_SHARED_LIBRARY_NAME="libtensorflow_framework.so.2" +build --action_env TF_NEED_CUDA="0" +build --config=manylinux2010cuda101 +test --config=manylinux2010cuda101 diff --git a/setup.py b/setup.py index 69245d6..ed85d22 100644 --- a/setup.py +++ b/setup.py @@ -23,9 +23,9 @@ from setuptools.dist import Distribution -__version__ = '0.0.1' +__version__ = '0.0.2' REQUIRED_PACKAGES = [ - 'tensorflow >= 2.1.0', + 'tensorflow == 2.11.0', ] project_name = 'tensorflow-custom-ops'