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