Skip to content
Open

Dev #301

Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Dockerfiles/BAZEL_VERSION
Original file line number Diff line number Diff line change
@@ -1 +1 @@
5.1.1
6.5.0
2 changes: 1 addition & 1 deletion Dockerfiles/install-ubuntu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ done

if $cuda; then
# install libcupti
apt-get -y install cuda-command-line-tools-11-6
apt-get -y install cuda-command-line-tools-12-3
fi

apt-get -y clean
Expand Down
2 changes: 1 addition & 1 deletion Dockerfiles/ubuntu-cuda
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04
FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04

# copy the contents of this repository to the container
COPY . tensorflow_cc
Expand Down
2 changes: 1 addition & 1 deletion tensorflow_cc/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ set(LOCAL_RAM_RESOURCES 4096 CACHE STRING "The amount of local RAM resources pas
set(LOCAL_CPU_RESOURCES HOST_CPUS CACHE STRING "The amount of local CPU cores passed to bazel scheduler (e.g., 2).")
set(TENSORFLOW_TAG "v${version}" CACHE STRING "The tensorflow release tag to be checked out (default v${version}).")
option(INSTALL_PROTOBUF "Install protobuf compatible with tensorflow version." OFF)
set(CMAKE_CXX_STANDARD 14 CACHE STRING "The C++ standard for building and linking the library (e.g., 14).")
set(CMAKE_CXX_STANDARD 20 CACHE STRING "The C++ standard for building and linking the library (e.g., 14).")

# -------------
# CMake Options
Expand Down
2 changes: 1 addition & 1 deletion tensorflow_cc/PROJECT_VERSION
Original file line number Diff line number Diff line change
@@ -1 +1 @@
2.9.0
2.15.0
6 changes: 3 additions & 3 deletions tensorflow_cc/cmake/build_tensorflow.sh.in
Original file line number Diff line number Diff line change
Expand Up @@ -88,9 +88,9 @@ fi

# set bazel options for c++17 standard if necessary
cxx_std_opts=""
if [ "@CMAKE_CXX_STANDARD@" == 17 ]; then
echo "Using C++17 standard"
cxx_std_opts="--config=c++17_gcc"
if [ "@CMAKE_CXX_STANDARD@" == 20 ]; then
echo "Using C++20 standard"
cxx_std_opts="--config=c++20_gcc"
fi

# configure and build
Expand Down
2 changes: 1 addition & 1 deletion ubuntu-requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ if [[ "$OSTYPE" == "linux-gnu" ]]; then
apt-get -y update
apt-get -y install \
cmake \
g++-9 \
g++ \
git \
python3-dev \
python3-numpy \
Expand Down