原因:colab中cuda版本与安装cuda版本不一致。
解决方案:将cuda11.2降级到10.2
本人参考案例
预期目标:!conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=10.2 -c pytorch后安装apex
colab环境:Ubuntu 18.04.6 LTS, cuda 11.2
apex安装方法:
代码框1:%%writefile setup.sh
git clone https://github.com/NVIDIA/apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./apex
代码框2:!sh setup.sh