InternVL 官网
interVL 安装
完全删除环境和环境中的所有软件包
conda remove -n env_name --all
安装
-
克隆此存储库:
git clone https://github.com/OpenGVLab/InternVL.git
-
创建conda虚拟环境并激活:
conda create -n inter python=3.9 -y
conda activate inter
-
安装
PyTorch>=2.0
并torchvision>=0.15.2
使用CUDA>=11.6
:例如,要
torch==2.0.1
安装CUDA==11.8
:
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
# or
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
-
安装
flash-attn==2.3.6
:
pip install flash-attn==2.3.6 --no-build-isolation
-
安装
timm==0.9.12
并mmcv-full==1.6.2
:
pip install timm==0.9.12
pip install -U openmim
mim install mmcv-full==1.6.2 # (optional, for mmsegmentation)
-
安装
transformers==4.37.2
:
pip install transformers==4.37.2
-
安装其他要求:
pip install opencv-python termcolor yacs pyyaml scipy
pip install deepspeed==0.13.5
pip install pycocoevalcap tqdm