Update Dockerfile.tf

nubispc · Mar 22, 2023 · 56921cc · 56921cc
1 parent f3b7949
commit 56921cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.tf b/Dockerfile.tf
@@ -141,8 +141,8 @@ RUN cd /tensorflow \
 # Build TensorFlow with bazel
 # https://github.com/nubispc/container-build-example/blob/86d84a184f90781c3eb2d47c87502fe14ba50361/Dockerfile.tensorflow-plugin#L106
 RUN cd /tensorflow \
-    && bazel build --local_ram_resources=HOST_RAM*.7 \
-            --local_cpu_resources=HOST_CPUS-3 \
+    && bazel build --local_ram_resources=HOST_RAM*.9 \
+            --local_cpu_resources=HOST_CPUS-1 \
             --config=v2 \
             --copt=-O3 \
             --config=opt \