mxnet gpu 版本安装
由于安装 mxnet gpu 版本踩了不少的坑 所以这里将踩的坑简单的记录下来
安装 mxnet gpu 版本之前需要先安装 cuda 以及cudnn 安装过程见我的博客中《Linux 深度学习服务器搭建》这里我安装的是cuda_10.1.168_418.67_linux.run 版本已经 对应的 cudnn-10.1-linux-x64-v7.6.5.32.tar 。
安装对应版本的 mxnet gpu 版本 , 安装命令如下:
pip install mxnet-cu101==1.7.0
注意:这里一定要安装对应的版本 否则会报各种错误