| name: llava_env # 环境名称 | |
| channels: | |
| - defaults # 使用默认的 channels（如有特殊需求，可加入其他 channels） | |
| dependencies: | |
| - python=3.11 # Python 版本 | |
| # 安装通过 pip 安装的 Python 库 | |
| - pip | |
| - pip: | |
| - torch==2.0.1 | |
| - accelerate==0.21.0 | |
| - bitsandbytes==0.41.0 | |
| - deepspeed==0.9.5 | |
| - einops-exts==0.0.4 | |
| - einops==0.6.1 | |
| - gradio==3.35.2 | |
| - gradio_client==0.2.9 | |
| - httpx==0.24.0 | |
| - markdown2==2.4.10 | |
| - numpy==1.26.0 | |
| - peft==0.4.0 | |
| - scikit-learn==1.2.2 | |
| - sentencepiece==0.1.99 | |
| - shortuuid==1.0.11 | |
| - timm==0.6.13 | |
| - tokenizers==0.13.3 | |
| - torchvision==0.15.2 | |
| - transformers==4.31.0 | |
| - wandb==0.15.12 | |
| - wavedrom==2.0.3.post3 | |
| - Pygments==2.16.1 | |
| - opencv-python | |
| - pycocotools | |
| # 如果需要添加运行命令或其他脚本执行步骤，可以在 run 中定义 | |