保姆级Colab上跑通SwinTransformer:
免费使用GPU资源
花费了两周,踩了无数的坑,终于跑通了,下面介绍几种方案:
第一种:目前Colab上的cuda版本最新版是12.2,降低cuda版本到11.x
第二种:使用最新版的cuda12.2,修改项目代码
最后附上遇到的各种问题的解决方法。
注意:swintransformer官方发布的代码,mmcv的版本是
mmcv>=1.2.4, <=1.4.0.
所以提前找好相应的cuda,torch版本,后面就不要安装错了再改了,一次安装成功。
方案一:降低cuda版本至11.1
跑通swintransformer最关键的方法是版本一定要对照。否则各种问题层出不穷。
我的版本:
cuda:11.1
python:3.8
torch1.8.2 torchvision0.9.2 torchaudio==0.8.2
apex:0.1(22.04dev)
mmcv-full:1.3.17
mmdet:2.11.0
step1:降级colab的版本:
到英伟达官网上下载源码编译安装
!wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb
!apt-key add /var/cuda-repo-ubuntu1804-11-1-local/7fa2af80.pub
!apt-get update
!apt-get install cuda-11-1
!export CUDA_PATH=/usr/local/cuda-11.1
!nvcc --version
结果:
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2020 NV