【无标题】

本文提供了一种在Colab上配置SwinTransformer环境的详细步骤,包括降低CUDA版本到11.1,安装特定版本的Python、Torch、MMCV和MMDetection。通过解决各种依赖和版本匹配问题,最终成功运行SwinTransformer模型。
摘要由CSDN通过智能技术生成

保姆级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

  • 7
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值