安装流程
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安装成功