Update Dockerfile

Dockerfile

@@ -91,7 +91,12 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install --upgrade bitsandbytes
 
 
-RUN pip install --extra-index-url https://pypi.ngc.nvidia.com nvidia-tensorflow>=0.0.1.dev5
+RUN pip install --user nvidia-tensorflow[horovod]
+
+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
+