解决方案:

git clone https://github.com/NVIDIA/apex

cd apex

python setup.py install

在使用python setup.py install 的过程中会报错, 错误截图

解决方案:

需要在命令行执行 export TORCH_CUDA_ARCH_LIST="6.0" 随便一下计算力值输入即可.然后在执行python setup.py install 即可以安装成功.

啊.....总算安装成功, 在这里卡了好长, 心酸...... 大腿都拍肿了!!!

更多推荐