mmyolo安装过程

mmyolo里最麻烦的就是mmcv,版本纷繁复杂,而且必须的对上才能安装成功。

以下是步骤总结:

  1. 首先得有台带GPU的机器,我手上是花两千多买的二手挖矿机平台,配上手里原来的CPU,内存,硬盘阵列,再买四张P40,插上,总共额外花费五千左右。
  2. 做raid5,安装Ubuntu,很简单,这里省略。
  3. 安装NVIDIA驱动,驱动是按照官网上的版本下载安装的,官网地址:
https://www.nvidia.com/Download/index.aspx?lang=en

image
点搜索跳转,下载驱动,安装即可

https://www.nvidia.com/Download/driverResults.aspx/160640/en-us/

我这里用CUDA11.8

  1. 安装CUDA
    CUDA官网下载列表
    安装完验证一下是否安装成功,使用命令:
nvcc -V

CuDNN不用着急安装,待会儿命令pip直接装。
到此为止,并行计算的机器安装好了。
下面开始Python 人工智能的环境安装,人工智能环境需要根据需要分不同的虚拟环境,所以需要conda,个人觉得conda管理虚拟环境比pip好用。

  1. 安装anaconda。
    请参考这篇文章Ubuntu20安装annaconda
    安装完成以后创建虚拟环境,老版本的用
conda create 虚拟环境名称
source activate 虚拟环境名称

这时候命令行左边会出现虚拟环境名称

  1. 可以开始安装pytorch
    但是这时候不要直接到官网随便安装一个版本,因为后面的软件决定了用哪个版本的pytorch, 以MMyolo为例,跟mmcv兼容的pytorch是2.0.x,而不是最新版。
    以终为始,我们需要先安装pytorch2.0 Pytorch历史版本
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
  1. 安装mmyolo. 按照官网文档安装依赖。
    mmyolo安装
posted @ 2024-02-22 10:49  williamyuan  阅读(455)  评论(0)    收藏  举报