colab配置mmdetection环境_by2021.6

最新推荐文章于 2023-05-21 22:20:13 发布

hoesen

最新推荐文章于 2023-05-21 22:20:13 发布

阅读量3.2k

点赞数 8

分类专栏： mmdetection colab 文章标签： gpu cuda 深度学习 pytorch

本文链接：https://blog.csdn.net/weixin_45902204/article/details/117956905

版权

本文介绍了在Colab环境中配置mmdetection时遇到CUDA版本与PyTorch不兼容的问题，以及如何解决。作者详细展示了检查系统信息、卸载旧版PyTorch、安装兼容CUDA11.0的PyTorch和mmdetection、以及处理安装后需重启Runtime的提示的步骤。最后，验证了安装成功并提供了运行mmdetection的基本检查。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文基于colab环境下，配置mmdetection环境，配置下来也踩了不少坑，希望能对有需要的朋友一点点帮助。

安装库最常见的bug大都是由于版本不兼容导致的（软件和硬件不兼容、不同库之间有冲突），所以在安装前要先确认平台的基本信息：平台系统、cuda版本、GCC版本。

#输入
!sudo lsb_release -a # 查看系统版本
!nvcc -V #或者!nvidia-smi # 查看cuda版本
!gcc --version # 查看GCC版本
——————————————————手动分割线————————————————
#输出
No LSB modules are available.
Distributor ID:	Ubuntu
Description:	Ubuntu 18.04.5 LTS
Release:	18.04
Codename:	bionic
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_Jul_22_19:09:09_PDT_2020
Cuda compilation tools, release 11.0, V11.0.221
Build cuda_11.0_bu.TC445_37.28845127_0
gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
Copyright (C) 2017 Free Software Foundation, Inc.
This is free software; see the source for copying conditions.  There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

可知：系统是Ubuntu18.04，cud