Update Dockerfile

Dockerfile

@@ -66,8 +66,6 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
     python3 -m pip install --upgrade pip
 
 
-RUN curl -L https://github.com/bazelbuild/bazelisk/releases/latest/download/bazelisk-linux-amd64 -o bazel && \
-    chmod +x bazel
 
 # ---------------- PyTorch cu128 (pinned) ----------------
 RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
@@ -97,9 +95,6 @@ RUN pip install --upgrade bitsandbytes
 RUN pip install --upgrade pip setuptools wheel
 
 RUN pip install nvidia-pyindex
-RUN bazel build -c opt --config=cuda --cxxopt=-D_GLIBCXX_USE_CXX11_ABI=0 tensorflow/tools/pip_package:build_pip_package
-RUN bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/pip --gpu --project_name tensorflow
-RUN pip install --no-cache-dir --upgrade /tmp/pip/tensorflow-*.whl