(1)
首先,如果有管理员权限的话,建议按照
安装mxnet-gpu版,解决在import时报错“OSError: libnccl.so.2“的问题_WangNJ1998的博客-CSDN博客
和
Installation Guide :: NVIDIA Deep Learning NCCL Documentation
安装ncll,问题应该是可以解决的。
(2)
如果像我一样没有sudo权限,建议安装低版本的MXnet-cu(虽然运行时会有警告,但可以不管的,或者按提示set MXNET_CUDA_LIB_CHECKING=0也可以)
推荐安装方法及版本
conda create -c anaconda -c conda-forge -n NAME python=3.6 cudatoolkit=9.2 cudnn=7.2
配置好环境后,
conda activate NAME
pip install mxnet-cu92
即可
运行成功后,进入python
import mxnet as mx
查看是否报错
(3)
运行时可能会有一种报错如下
MXNetError: Check failed: compileResult == NVRTC_SUCCESS (5 vs. 0) : NVRTC Compilation failed. Please set environment variable MXNET_USE_FUSION to 0.
只需在运行程序前,在命令行键入
export MXNET_USE_FUSION=0
即可