Update Dockerfile

Dockerfile

@@ -91,8 +91,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN pip install --upgrade bitsandbytes
 
 
-RUN pip install --user nvidia-tensorflow[horovod]
-
+RUN pip install nvidia-pyindex
+RUN pip install 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