安装apex:
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install
安装过程中出现问题:TypeError unsupported operand type(s) for +: “NoneType” and “str”
造成这种原因的因素可能很多,一种尝试是:
先安装:
export TORCH_CUDA_ARCH_LIST="compute capability"
conda install -c nvidia cuda-nvcc
安装成功后,再尝试:
python setup.py install
安装成功。
版权声明:本文为qq_33854260原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。