Updating LD_LIBRARY_PATH for tf-2.0 Dockerfile.gpu (#293)

* Updating LD_LIBRARY_PATH for Dockerfile.gpu

Adding `/usr/local/cuda/extras/CUPTI/lib64` to LD_LIBRARY_PATH to include required CUDA  *.so files.

* Updating LD_LIBRARY_PATH for tf-2.1 py3 Dockerfile.gpu

Added "/usr/local/cuda/extras/CUPTI/lib64" to LD_LIBRARY_PATH for access to additional libs

Co-authored-by: Saravana Kumar <[email protected]>

Author: preshen-goobiah

docker/2.0.0/py3/Dockerfile.gpu

@@ -106,7 +106,7 @@ RUN echo "hwloc_base_binding_policy = none" >> /usr/local/etc/openmpi-mca-params
 # Set default NCCL parameters
 RUN echo NCCL_DEBUG=INFO >> /etc/nccl.conf
 
-ENV LD_LIBRARY_PATH=/usr/local/openmpi/lib:$LD_LIBRARY_PATH
+ENV LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:/usr/local/openmpi/lib:$LD_LIBRARY_PATH
 ENV PATH /usr/local/openmpi/bin/:$PATH
 ENV PATH=/usr/local/nvidia/bin:$PATH
 

docker/2.1.0/py3/Dockerfile.gpu

@@ -109,7 +109,7 @@ RUN echo "hwloc_base_binding_policy = none" >> /usr/local/etc/openmpi-mca-params
 # Set default NCCL parameters
 RUN echo NCCL_DEBUG=INFO >> /etc/nccl.conf
 
-ENV LD_LIBRARY_PATH=/usr/local/openmpi/lib:$LD_LIBRARY_PATH
+ENV LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:/usr/local/openmpi/lib:$LD_LIBRARY_PATH
 ENV PATH /usr/local/openmpi/bin/:$PATH
 ENV PATH=/usr/local/nvidia/bin:$PATH