安装mxnet-gpu 版本遇到的问题
遇到这个问题上的时候:
oserror: libcudart.so.10.0: cannot open shared object file: no such file or directory
解决办法:
1、先搞清楚你安装的mxnet版本和cuda的版本是否对上号
例如:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple mxnet-cu100
安装的是cuda10.0的mxnet 版本,如果你在 /usr/local/下找不到cuda10.0,不用说,你的cuda版本没有对上,nvcc -V 查看版本,再安装mxnet对应的GPU版本
nvidia-smi
我查看我的cuda版本cuda10.2,所以我的命令:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple mxnet-cu102
安装成功
import mxnet as mx
没有再报错
2、还有一种情况,你找到了对应的cuda10.0,那么你就要这样做一下:
sudo ldconfig /usr/local/cuda-10.0/lib64
应该可行