Update Dockerfile

Dockerfile

@@ -93,7 +93,6 @@ RUN pip install --upgrade bitsandbytes
 RUN pip install --upgrade pip setuptools wheel
 
 RUN pip install nvidia-pyindex
-RUN pip install --extra-index-url=https://pypi.ngc.nvidia.com --trusted-host pypi.ngc.nvidia.com nvidia-tensorflow
 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