安装mxnet报错OSError: libcudnn.so.8

安装流程

pip install mxnet-cu102

安装好了之后在python中import mxnet,报了OSError: libcudnn.so.8: cannot open shared object file: No such file or directory错误

解决方案

方案1

改变mxnet的版本,卸载之后重新尝试了mxnet-cu100和mxnet-cu110,依然报相同的错误,未解决

方案2

网上看了很多解决方案,因为没有sudo权限所以没有尝试

官网上看到docker的安装方法所以尝试

首先要安装NVIDIA的docker驱动,然后拉取镜像docker pull mxnet/python:gpu,mxnet/python 的docker hub里只有1.9版本的镜像,但在拉取镜像的过程中报错

这样可以选择拉取别人上传的镜像也可以拉取一个linux镜像重新搭建需要的cuda和cuDNN版本

拉取镜像构建容器参考Docker容器构建

方案3

最后的解决方法有点简单,兜兜转转一圈回来下载一个低版本的安装便成功了

pip install mxnet-cu102==1.6.0.post0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

直接pip install mxnet-cu102安装的版本是1.9.1会报错,1.6.0版本没有报错

其他方案

在虚拟环境下装个10.1 cudatookit

先执行conda install cudatoolkit=10.1 -c pytorch
再pip install mxnet-cu101==1.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/

上述方案也可以将mxnet安装成功

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注