我正在尝试安装 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
它能做什么:
- 下载并安装 Lambda Stack 存储库(本质上将文件添加到 /etc/apt/sources.list.d/)
- 更新 apt 并安装
lambda-stack-cuda
软件包。 - 将 CUDA、驱动程序、CuDNN 和具有 CuDNN 和 GPU 支持的 TensorFlow 安装到适当的系统级目录中。您无需修改
LD_LIBRARY_PATH
或,PATH
因为共享库位于ld
已在链接时检查的目录中。