Ubuntu20.04下RStudio运行GPU版Tensorflow找不到libcudart.so.11.0

问题描述 #

需要在Ubuntu 20.04 LTS系统的RStudio中运行支持GPU加速的TensorFlow。按照RStudio官方安装指南完成以下操作：

• 安装Nvidia 470版本驱动、CUDA 11.2、cuDDN 8.1.0
• 修改~/.bashrc添加如下环境变量配置：

export CUDA_HOME=/usr/local/cuda
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH

• 执行install_keras(tensorflow = "gpu")安装GPU版本的TensorFlow

运行时出现如下报错：

W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /home/nickopotamus/.local/share/r-miniconda/envs/r-reticulate/lib:/usr/lib/R/lib::/lib:/usr/lib/x86_64-linux-gnu:/usr/lib/jvm/default-java/lib/server

排查过程 #

执行sudo find / -name 'libcudart.so.11.0'查询到该文件存在于以下路径：

/home/nickopotamus/.local/share/r-miniconda/envs/r-reticulate/lib/libcudart.so.11.0
/home/nickopotamus/anaconda3/pkgs/cudatoolkit-11.3.1-h2bc3f7f_2/lib/libcudart.so.11.0
/home/nickopotamus/anaconda3/pkgs/cudatoolkit-11.2.0-h73cb219_8/lib/libcudart.so.11.0
/home/nickopotamus/anaconda3/pkgs/cudatoolkit-11.2.72-h2bc3f7f_0/lib/libcudart.so.11.0
/usr/local/cuda-11.2/targets/x86_64-linux/lib/libcudart.so.11.0

报错信息的动态库搜索路径列表中已经包含上述第一条路径，初步怀疑是多余的Anaconda包导致版本冲突。

最终解决方案 #

通过NVIDIA Data Science Stack工具从头安装所有相关组件，故障解决。

内容的提问来源于stack exchange，提问作者Nickopotamus