import tensorflow cannot find libcudart.so.11.0 despite cuda-cudart being installed

Viewed 1167

I have a Conda environment with cuda, cudnn, cudatoolkit installed from the nvidia channel and tensorflow installed from PyPI.

When I try import tensorflow, I get

could not load dynamic library 'libcudart.so.11.0';

even though it is in $CONDA_PREFIX/lib.

I can work around this by appending/prepending this path to LD_LIBRARY_PATH manually (shared objects are in lib directory). I know this can be automated via a script in opt/conda/activate.d/ but shouldn't conda activate env-name resolve this automatically?

Is something wrong with my Conda installation / environment?


Configuration Details

Python version

Python 3.9.7 (default, Sep 16 2021, 13:09:58) 
[GCC 7.5.0] :: Anaconda, Inc. on linux

Relevant Conda Packages

$ conda list -n env-name '(cuda|tensorflow)'
# Name                    Version                   Build  Channel
cuda                      11.5.0               hc28fa2a_0    nvidia
cuda-cccl                 11.5.62                       0    nvidia
cuda-command-line-tools   11.5.0               hc28fa2a_0    nvidia
cuda-compiler             11.5.0               hc28fa2a_0    nvidia
cuda-cudart               11.5.50              h79feb7f_0    nvidia
cuda-cudart-dev           11.5.50              h2215e51_0    nvidia
cuda-cuobjdump            11.5.50              h3fb44dc_0    nvidia
cuda-cupti                11.5.57              h2e3e70b_0    nvidia
cuda-cuxxfilt             11.5.50              he5660e7_0    nvidia
cuda-driver-dev           11.5.50                       0    nvidia
cuda-gdb                  11.5.50              h03b1d8f_0    nvidia
cuda-libraries            11.5.0               hc28fa2a_0    nvidia
cuda-libraries-dev        11.5.0               hc28fa2a_0    nvidia
cuda-memcheck             11.5.50              h29df99e_0    nvidia
cuda-nvcc                 11.5.50              h8f81028_0    nvidia
cuda-nvdisasm             11.5.50              hc026f5c_0    nvidia
cuda-nvml-dev             11.5.50              h511b398_0    nvidia
cuda-nvprof               11.5.50              h1597bcb_0    nvidia
cuda-nvprune              11.5.50              h003a74f_0    nvidia
cuda-nvrtc                11.5.50              h895d16b_0    nvidia
cuda-nvrtc-dev            11.5.50              h2215e51_0    nvidia
cuda-nvtx                 11.5.50              h4855407_0    nvidia
cuda-nvvp                 11.5.50              h97cf613_0    nvidia
cuda-runtime              11.5.0               hc28fa2a_0    nvidia
cuda-samples              11.5.56              hf1e648b_0    nvidia
cuda-sanitizer-api        11.5.50              h7b563e8_0    nvidia
cuda-toolkit              11.5.0               hc28fa2a_0    nvidia
cuda-tools                11.5.0               hc28fa2a_0    nvidia
cuda-visual-tools         11.5.0               hc28fa2a_0    nvidia
cudatoolkit               11.5.0               h36ae40a_9    nvidia
tensorflow                2.7.0                    pypi_0    pypi
tensorflow-estimator      2.7.0                    pypi_0    pypi
tensorflow-hub            0.12.0                   pypi_0    pypi
tensorflow-io-gcs-filesystem 0.21.0                   pypi_0    pypi
0 Answers
Related