下载apex
git clone https://github.com/NVIDIA/apex.git
安装失败出现
RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries. Pytorch binaries were compiled with Cuda XXXX
问题时:
说明安装的cuda和pytorch版本下的cuda版本不一致;整一致了……
2. 解决了上面问题,就可以继续安装了,如下图:然后出来一长溜
就安装成功啦