NVIDIA / tensorflow

An Open Source Machine Learning Framework for Everyone
https://developer.nvidia.com/deep-learning-frameworks
Apache License 2.0
962 stars 144 forks source link

Tensorrt 7.2.2 install issues #75

Closed fabiosanger closed 1 year ago

fabiosanger commented 1 year ago

Please make sure that this is a build/installation issue. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:build_template

System information

Describe the problem According to tensorflow I should be installing tensorrt=7.2.2 but i fail to find the correct repository I tried to install via pip, but the available version are > 8

i get the following warning when importing tensorflow

Could not load dynamic library 'libnvinfer.so.7' Could not load dynamic library 'libnvinfer_plugin.so.7'

fabiosanger commented 1 year ago

I got it working as follow:

RUN pip install tensorflow==2.11.0 \
    tensorboard==2.11.0 \
    jupyterlab \
    jupyter_client==7.0.6 \
    nvidia-pyindex \
    nvidia-tensorrt==7.2.3.4
RUN pip install --pre jupyter-tensorboard

RUN echo 'export LD_LIBRARY_PATH=/opt/conda/envs/${ENV_NAME}/lib/${PYTHON_VERSION}/site-packages/tensorrt/:$LD_LIBRARY_PATH' >> ~/.bashrc