装完cuda和cudnn后,安装mxnet-cu91后,import mxnet的时候报错libcudart.so.9.1: cannot open shared object file: No such file or directory;
解决:
32-bit:
sudo ldconfig /usr/local/cuda/lib
64-bit:
sudo ldconfig /usr/local/cuda/lib64
注意自行选择你的机子对应的版本
然后再次import时虽然可以import成功但是有报
FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
from ._conv import register_converters as _register_converters
这是h5py包的问题
解决:
对h5py进行更新升级即可
pip install h5py==2.8.0rc1
Done!