【6.6】OpenMMLab实战营课程笔记⑥

MMPretrain相关安装

一、安装环境准备

（提前安装好python、cuda和pytorch）

conda activate openmmlab

进入python交互环境

import torch
print(torch.__version__)
print(torch.cuda.is_available())
exit()

回复是，，目前本机的pytorch的版本是2.0.0，pytorch也可进行cuda支持。

安装环境符合要求。

二、下载并安装

下载mmpretrain仓库：

git clone https://github.com/open-mmlab/mmpretrain

安装openmim

cd mmpretrain
pip install openmim 

 用openmim安装mmpretrain算法库，放在mmpretrain路径下

mim install -e ".[multimodal]"

到这一步，报错了……

ERROR: No .egg-info directory found in C:\Users\liuzh\AppData\Local\Temp\pip-pip-egg-info-tlpdgjbv

看了很多帖子，大多是说版本不匹配造成的，还在解决问题，后面再继续更新吧~

【6.8】继续更新……

上面问题的解决方案：重新搭建了新的虚拟环境，换了和老师一模一样的python版本3.10.11，同时重新下载了新的pytorch文件2.0.1+cu117，问题就解决了。

确认MMPretrain的版本

import mmpretrain
print(mmpretrain.__version__)

目前，本机的版本是：1.0.0rc8