美文网首页
Linux安装MXNet

Linux安装MXNet

作者: haoshengup | 来源:发表于2022-04-25 17:02 被阅读0次

  最近安装MXNet碰到了一些问题,这里记录一下解决方案。

  1、通过官网的给的pip方式安装,安装之后导入出现错误Illegal instruction (core dumped)。更换了几个版本(1.6, 1.7, 1.8)都报同样的错误。查找之后发现,是因为mxnet pip安装支持编码集是avx,但是当前的服务器不支持。
  可以通过cat /proc/cpuinfo查看支持的编码集:

flags           : fpu vme de pse tsc msr pae mce cx8 apic sep mtrr pge mca cmov pat pse36 clflush dts acpi mmx fxsr sse sse2 ss ht tm pbe syscall nx pdpe1gb rdtscp lm constant_tsc arch_perfmon pebs bts rep_good nopl xtopology nonstop_tsc cpuid aperfmperf pni pclmulqdq dtes64 monitor ds_cpl vmx smx est tm2 ssse3 cx16 xtpr pdcm pcid dca sse4_1 sse4_2 x2apic popcnt tsc_deadline_timer aes xsave avx lahf_lm epb pti ssbd ibrs ibpb stibp tpr_shadow vnmi flexpriority ept vpid xsaveopt dtherm ida arat pln pts md_clear flush_l1d

  解决方案:本来打算通过源码编译安装。但是,试了试最新的1.9版本,居然导入成功,这样节省了很多时间。

  2、安装之后在控制台可以导入MXNet成功



  但是,在pycharm中使用的时候会报如下错误:

OSError: /home/djzheng/anaconda3/envs/insightface/lib/python3.7/site-packages/mxnet/libmxnet.so: symbol cudnnDestroyRNNDataDescriptor, version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference

  从报错的信息来看,是因为加载cudnn库失败。但是在控制台终端可以加载,说明环境存在,但是pycharm没加载到。于是,查看当前conda虚拟环境加载了哪些环境变量,发现cuda的环境变量已经被加载:



  然后,问题找到这里就被卡住了。网上找了一些资料,也都是说cuda的环境问题。但是这里已经被加载了。折腾了很长时间,突然发现报错信息里的描述version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference,说是加载不到cudnn7.0。到这里终于找到问题,是cudnn的版本问题。因为机器上装了多个版本的cuda,而这里的/usr/local/cuda/binusr/local/cuda/lib64都只是软链接,指向当前使用的cuda版本(CUDA 10.1,cuDNN v8.0.5)。也就是说cudnn的版本不匹配。
  从这里可以推测出,mxnet-1.9.0的cuda10.1版本需要的cudnn版本是7.0。然后,发现机器上安装的cuda10.0用的是cudnn7.0。所以,这里直接修改pycharm中加载的LD_LIBRARY_PATH目录,让其指向cudnn7.0:

  修改之后,重新import,发现可以加载成功。

参考:
http://www.4k8k.xyz/article/jinhao_2008/104060814
https://mxnet.apache.org/versions/1.9.0/get_started?version=v1.9.0&platform=linux&language=python&processor=gpu&environ=pip&
https://www.likecs.com/show-203509918.html
https://blog.csdn.net/uniqueyyc/article/details/88122204
https://discuss.gluon.ai/t/topic/2599

相关文章

  • MX-net+tensorflow

    MxNet-1.2.0(Linux-python-GPU-pip) 安装Tensorflow 卸载Tensorfl...

  • Linux安装MXNet

      最近安装MXNet碰到了一些问题,这里记录一下解决方案。   1、通过官网的给的pip方式安装,安装之后导入出...

  • 从0搭建MXNet环境

    01 目标 在没有Linux环境的前提下,从头开始安装Linux环境与cuda 并且编译安装mxnet的gpu加速...

  • 2019-02-19

    Ubuntu16.04 源码安装MXNet [参考官方教程](https://mxnet.incubator.ap...

  • 树莓派4B上mxnet目标检测环境搭建

    mxnet安装 先贴出官网的教程:https://mxnet.apache.org/versions/1.7.0/...

  • [MXnet] Simple Introduction to t

    MXnet的学习笔记,这次主要是MXnet提供的example的综述介绍。关于MXnet在OSX下的编译安装,可以...

  • Centos7用virtualenv pip安装mxnet

    很奇怪网上安装centos7+mxnet 都是自己编译。如《Centos7下安装部署MXNET》 问题及尝试 在v...

  • 优质学习资料

    不断筛选优秀资料: MXNet 中文教程:图像分类 用MXnet实战深度学习之一:安装GPU版mxnet并跑一个M...

  • 安装MXNet

    这里只介绍一下Mac下的安装记录,更多的请参考:https://mxnet.apache.org/install/...

  • MXNet安装

    mxnet学习笔记,安装. Note:This blog without all the resources th...

网友评论

      本文标题:Linux安装MXNet

      本文链接:https://www.haomeiwen.com/subject/voofyrtx.html