diff --git a/Dockerfiles/install-ubuntu.sh b/Dockerfiles/install-ubuntu.sh index c9caa39..c22c5a5 100755 --- a/Dockerfiles/install-ubuntu.sh +++ b/Dockerfiles/install-ubuntu.sh @@ -51,7 +51,7 @@ if $shared; then fi if $cuda; then # install libcupti - apt-get -y install cuda-command-line-tools-9-2 + apt-get -y install cuda-command-line-tools-10-0 fi apt-get -y clean diff --git a/Dockerfiles/ubuntu-shared-cuda b/Dockerfiles/ubuntu-shared-cuda index 8ae9645..54bee88 100644 --- a/Dockerfiles/ubuntu-shared-cuda +++ b/Dockerfiles/ubuntu-shared-cuda @@ -1,4 +1,4 @@ -FROM nvidia/cuda:9.2-cudnn7-devel-ubuntu18.04 +FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 # copy the contents of this repository to the container COPY . tensorflow_cc diff --git a/README.md b/README.md index 285c49d..a690b31 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # tensorflow_cc [![Build Status](http://ash.floop.cz:8080/buildStatus/icon?job=tensorflow_cc)](http://ash.floop.cz:8080/job/tensorflow_cc/) -[![TF version](https://img.shields.io/badge/TF%20version-1.12.0-brightgreen.svg)]() +[![TF version](https://img.shields.io/badge/TF%20version-1.13.0-brightgreen.svg)]() This 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. diff --git a/tensorflow_cc/CMakeLists.txt b/tensorflow_cc/CMakeLists.txt index 1bd4e98..da7576c 100644 --- a/tensorflow_cc/CMakeLists.txt +++ b/tensorflow_cc/CMakeLists.txt @@ -1,7 +1,7 @@ cmake_minimum_required(VERSION 3.3 FATAL_ERROR) project( "tensorflow_cc" - VERSION 1.12.0 + VERSION 1.13.0 ) # Static library with no GPU support is built by default. @@ -10,7 +10,7 @@ project( option(TENSORFLOW_SHARED "Build shared library (required for GPU support)." OFF) option(ALLOW_CUDA "When building the shared library, try to find and use CUDA." ON) option(TENSORFLOW_STATIC "Build static library." ON) -set(TENSORFLOW_TAG "c985bd0dce0f8a7ccf334c9782d051c81ad00f1d" CACHE STRING "The tensorflow release tag to be checked out (default v1.12.0).") +set(TENSORFLOW_TAG "v1.13.0-rc1" CACHE STRING "The tensorflow release tag to be checked out (default v1.13.0).") option(SYSTEM_PROTOBUF "Use system protobuf instead of static protobuf from contrib/makefile." OFF) # -------------