no module named ‘apex‘

官方安装

'''
我的环境
Linux-Ubuntu 16.04.7 LTS
python=3.7
pytorch=1.4
cuda=10.1
'''
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
'''
三步走，安装失败
ERROR: Command errored out with exit status 1: /home/ty19/miniconda3/envs/saner/bin/python -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-2ex7wchp/setup.py'"'"'; __file__='"'"'/tmp/pip-req-build-2ex7wchp/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' --cpp_ext --cuda_ext install --record /tmp/pip-record-uoab470z/install-record.txt --single-version-externally-managed --compile --install-headers /home/ty19/miniconda3/envs/saner/include/python3.7m/apex Check the logs for full command output.

pip._internal.exceptions.InstallationSubprocessError: Command errored out with exit status 1: /home/ty19/miniconda3/envs/saner/bin/python -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-2ex7wchp/setup.py'"'"'; __file__='"'"'/tmp/pip-req-build-2ex7wchp/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' --cpp_ext --cuda_ext install --record /tmp/pip-record-uoab470z/install-record.txt --single-version-externally-managed --compile --install-headers /home/ty19/miniconda3/envs/saner/include/python3.7m/apex Check the logs for full command output.

'''

原因：
使用的pytorch版本为1.4
但是torchvision没有安装，再安装了torchvision=0.5之后，使用上述的官方安装方法成功安装apex。

pip install torchvision==0.5

注意：安装时会提示warning: integer conversion resulted in a change of sign

总结：
1、安装apex时，一定要注意torch、torchversion、cuda、linux版本对应
2、在查看cuda版本时一定要使用nvcc -V命令，不要使用nidia-smi命令，有时后者得到的版本是不准确的。