Do not copy libs on every datahub pod boot

ucsd-ets · Mar 4, 2024 · e51f49c · e51f49c
1 parent da9c653
commit e51f49c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -108,4 +108,10 @@ RUN mkdir -p $CONDA_DIR/lib/nvvm/libdevice && \
     cp $CONDA_DIR/nvvm/libdevice/libdevice.10.bc $CONDA_DIR/lib/nvvm/libdevice/
     #CUDA 11: cp $CONDA_DIR/lib/libdevice.10.bc $CONDA_DIR/lib/nvvm/libdevice/
 
+# TensorRT fix for tensorflow
+# https://github.com/tensorflow/tensorflow/issues/61468 (could not find TensorRT)
+# This will most definitely have to be changed after 8.6.1...
+RUN ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer_plugin.so.8 /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer_plugin.so.8.6.1 && \
+    ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.8 /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.8.6.1
+
 RUN . /tmp/activate.sh
diff --git a/images/scipy-ml-notebook/tensorrt_env_vars.sh b/images/scipy-ml-notebook/tensorrt_env_vars.sh
@@ -3,11 +3,6 @@
 # put tensorrt lib into LD_LIBRARY_PATH
 #export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt:$LD_LIBRARY_PATH"
 
-# https://github.com/tensorflow/tensorflow/issues/61468 (could not find TensorRT)
-# This will most definitely have to be changed after 8.6.1...
-ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer_plugin.so.8 /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer_plugin.so.8.6.1
-ln -s /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.8 /opt/conda/lib/python3.11/site-packages/tensorrt_libs/libnvinfer.so.8.6.1
-
 # tensorrt 8.6+
 export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt:$LD_LIBRARY_PATH"
 export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt_bindings:$LD_LIBRARY_PATH"