File tree Expand file tree Collapse file tree 6 files changed +9
-18
lines changed
Expand file tree Collapse file tree 6 files changed +9
-18
lines changed Original file line number Diff line number Diff line change 1- 4.2 .1
1+ 5.1 .1
Original file line number Diff line number Diff line change 1717
1818if $cuda ; then
1919 # install libcupti
20- apt-get -y install cuda-command-line-tools-10-1
20+ apt-get -y install cuda-command-line-tools-11-6
2121fi
2222
2323apt-get -y clean
Original file line number Diff line number Diff line change 1- FROM nvidia/cuda:11.1 -cudnn8-devel-ubuntu18 .04
1+ FROM nvidia/cuda:11.6.2 -cudnn8-devel-ubuntu20 .04
22
33# copy the contents of this repository to the container
44COPY . tensorflow_cc
Original file line number Diff line number Diff line change 11# tensorflow_cc
22[ ![ Build Status] ( http://oak.floop.cz:8080/buildStatus/icon?job=tensorflow_cc )] ( http://oak.floop.cz:8080/job/tensorflow_cc/ )
3- [ ![ TF version] ( https://img.shields.io/badge/TF%20version-2.8 .0-brightgreen.svg )] ( )
3+ [ ![ TF version] ( https://img.shields.io/badge/TF%20version-2.9 .0-brightgreen.svg )] ( )
44
55This repository makes possible the usage of the [ TensorFlow C++] ( https://www.tensorflow.org/api_docs/cc/ ) API from the outside of the TensorFlow source code folders and without the use of the [ Bazel] ( https://bazel.build/ ) build system.
66
Original file line number Diff line number Diff line change @@ -83,23 +83,14 @@ install(
8383 REGEX ".*Eigen.*"
8484)
8585install (
86- FILES "${CMAKE_CURRENT_BINARY_DIR} /tensorflow/bazel-bin/tensorflow/libtensorflow_cc.so "
86+ DIRECTORY "${CMAKE_CURRENT_BINARY_DIR} /tensorflow/bazel-bin/tensorflow/"
8787 DESTINATION lib
88- RENAME libtensorflow_cc.so.${PROJECT_VERSION_MAJOR}
89- OPTIONAL
88+ FILES_MATCHING PATTERN "libtensorflow_cc.so*"
9089)
9190install (
92- FILES "${CMAKE_CURRENT_BINARY_DIR} /tensorflow/bazel-bin/bin/tensorflow/libtensorflow_cc.so "
91+ DIRECTORY "${CMAKE_CURRENT_BINARY_DIR} /tensorflow/bazel-bin/bin/tensorflow/"
9392 DESTINATION lib
94- RENAME libtensorflow_cc.so.${PROJECT_VERSION_MAJOR}
95- OPTIONAL
96- )
97- # Make versioned symlinks to the library
98- install (CODE "execute_process( \
99- COMMAND ${CMAKE_COMMAND} -E create_symlink \
100- libtensorflow_cc.so.${PROJECT_VERSION_MAJOR} \
101- ${CMAKE_INSTALL_PREFIX} /lib/libtensorflow_cc.so \
102- )"
93+ FILES_MATCHING PATTERN "libtensorflow_cc.so*"
10394)
10495
10596# --------------------------
Original file line number Diff line number Diff line change 1- 2.8 .0
1+ 2.9 .0
You can’t perform that action at this time.
0 commit comments