云服务器环境配置找不到动态库问题

部署运行你感兴趣的模型镜像

问题描述

配置好tensorflow环境后,输入

python -c "import tensorflow as tf; print(tf.test.is_built_with_cuda()); print(tf.test.is_gpu_available())"

提示

 Could not load dynamic library 'libcudart.so.11.0';
 ....
 Skipping registering GPU devices...
False

原因

系统查找动态库的默认路径是:

  1. /lib/usr/lib(系统默认路径)
  2. /etc/ld.so.conf 配置的路径
  3. 环境变量 LD_LIBRARY_PATH 中的路径

而用 conda install cudatoolkit cudnn 时:

  • 它们并不会安装到 /usr/local/cuda(系统路径)
  • 而是安装在:
$CONDA_PREFIX/lib/

例如:

/root/miniconda3/envs/自定义环境/lib/

但这个路径默认不在系统的搜索路径列表里


解决办法:手动导入

通过这条命令:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/

查询当前库路径:

echo $LD_LIBRARY_PATH

永久解决办法

编辑 ~/.bashrc~/.bash_profile

echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib' >> ~/.bashrc

然后:

source ~/.bashrc

这样以后每次激活环境时,路径会自动生效

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值