如何找到libcudart.so的路径?

如何找到libcudart.so的路径?

我正在尝试安装 Tensorflow GPU 版本,但出现此问题。我已通过运行安装了 nvidia-cuda-toolkit

 sudo apt install nvidia-cuda-toolkit

下载成功了。但我找不到这个 libcudart.so

Please specify which gcc nvcc should use as the host compiler. [Default is /usr/bin/gcc]: /usr/bin/gcc
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to use system default]: 
Please specify the location where CUDA  toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/local/cuda
Invalid path to CUDA  toolkit. /usr/local/cuda/lib64/libcudart.so cannot be found

我该如何解决这个问题?

答案1

不确定这是否是最好的方法,但我遇到了同样的问题,而且这很有帮助。

sudo ln -s /usr/local/cuda/lib64 /usr

/usr使用以下方式验证链接:ls -l lib64

lib64 -> /usr/local/cuda/lib64

答案2

看起来,你导出了错误的路径。

因此,在终端上输入:sudo ldconfig /usr/local/cuda/lib64

ldconfig 创建必要的链接并缓存到在命令行指定的目录中找到的最新共享库

如果这不起作用,请尝试:export PATH=$PATH:/usr/local/cuda/bin 这将设置环境路径。

答案3

如果您使用的是 Ubuntu 16.04 或 Ubuntu 18.04 并想要安装支持 GPU 的 TensorFlow,则 Lambda Stack 存储库中有一个适用于该操作系统的 deb 包。

您可以使用以下行来安装存储库和包:

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdal.com/static/files/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && sudo apt-get install -y lambda-stack-cuda

它能做什么:

  1. 下载并安装 Lambda Stack 存储库(本质上将文件添加到 /etc/apt/sources.list.d/)
  2. 更新 apt 并安装lambda-stack-cuda软件包。
  3. 将 CUDA、驱动程序、CuDNN 和具有 CuDNN 和 GPU 支持的 TensorFlow 安装到适当的系统级目录中。您无需修改LD_LIBRARY_PATH​​或,PATH因为共享库位于ld已在链接时检查的目录中。

相关内容