119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
VideoLingo 快速安装脚本 - 使用清华大学镜像源
|
|||
|
|
解决PyTorch等大型包下载缓慢的问题
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import subprocess
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
from rich.console import Console
|
|||
|
|
from rich.panel import Panel
|
|||
|
|
from rich.progress import track
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
console = Console()
|
|||
|
|
|
|||
|
|
def run_command(cmd, description=""):
|
|||
|
|
"""执行命令并显示进度"""
|
|||
|
|
console.print(f"🚀 {description}", style="bold green")
|
|||
|
|
console.print(f"执行命令: {cmd}", style="dim")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
result = subprocess.run(cmd, shell=True, check=True,
|
|||
|
|
capture_output=True, text=True)
|
|||
|
|
console.print("✅ 完成", style="bold green")
|
|||
|
|
return True
|
|||
|
|
except subprocess.CalledProcessError as e:
|
|||
|
|
console.print(f"❌ 错误: {e}", style="bold red")
|
|||
|
|
console.print(f"输出: {e.stdout}", style="dim")
|
|||
|
|
console.print(f"错误: {e.stderr}", style="dim")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
console.print(Panel.fit("🎯 VideoLingo 快速安装 - 清华镜像版", style="bold blue"))
|
|||
|
|
|
|||
|
|
# 1. 首先配置pip使用清华镜像
|
|||
|
|
console.print("📦 配置pip使用清华大学镜像源...")
|
|||
|
|
pip_config_cmd = """
|
|||
|
|
mkdir -p ~/.pip
|
|||
|
|
cat > ~/.pip/pip.conf << EOF
|
|||
|
|
[global]
|
|||
|
|
index-url = https://pypi.tuna.tsinghua.edu.cn/simple/
|
|||
|
|
[install]
|
|||
|
|
trusted-host = pypi.tuna.tsinghua.edu.cn
|
|||
|
|
EOF
|
|||
|
|
"""
|
|||
|
|
run_command(pip_config_cmd, "配置pip镜像源")
|
|||
|
|
|
|||
|
|
# 2. 先停止当前的pytorch下载（如果正在进行）
|
|||
|
|
console.print("⏹️ 准备重新安装PyTorch...")
|
|||
|
|
|
|||
|
|
# 3. 安装PyTorch及相关组件（使用清华镜像）
|
|||
|
|
pytorch_packages = [
|
|||
|
|
"torch==2.0.0+cu118",
|
|||
|
|
"torchvision==0.15.0+cu118",
|
|||
|
|
"torchaudio==2.0.0+cu118"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
pytorch_cmd = f"""pip install {' '.join(pytorch_packages)} \\
|
|||
|
|
-i https://pypi.tuna.tsinghua.edu.cn/simple/ \\
|
|||
|
|
--extra-index-url https://download.pytorch.org/whl/cu118 \\
|
|||
|
|
--timeout 300"""
|
|||
|
|
|
|||
|
|
if not run_command(pytorch_cmd, "安装PyTorch CUDA版本"):
|
|||
|
|
console.print("⚠️ PyTorch安装失败，尝试CPU版本...", style="yellow")
|
|||
|
|
cpu_cmd = "pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple/"
|
|||
|
|
run_command(cpu_cmd, "安装PyTorch CPU版本")
|
|||
|
|
|
|||
|
|
# 4. 安装其他可能需要的依赖
|
|||
|
|
other_packages = [
|
|||
|
|
"transformers",
|
|||
|
|
"accelerate",
|
|||
|
|
"datasets",
|
|||
|
|
"tokenizers",
|
|||
|
|
"huggingface-hub",
|
|||
|
|
"safetensors",
|
|||
|
|
"numpy",
|
|||
|
|
"opencv-python",
|
|||
|
|
"pillow",
|
|||
|
|
"matplotlib",
|
|||
|
|
"scikit-learn",
|
|||
|
|
"pandas",
|
|||
|
|
"tqdm",
|
|||
|
|
"psutil",
|
|||
|
|
"ffmpeg-python",
|
|||
|
|
"whisper-openai",
|
|||
|
|
"openai",
|
|||
|
|
"anthropic"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
console.print("📦 安装其他常用AI/ML依赖包...")
|
|||
|
|
for package in track(other_packages, description="安装依赖包..."):
|
|||
|
|
cmd = f"pip install {package} -i https://pypi.tuna.tsinghua.edu.cn/simple/ --timeout 180"
|
|||
|
|
subprocess.run(cmd, shell=True, capture_output=True)
|
|||
|
|
time.sleep(0.1) # 避免过快请求
|
|||
|
|
|
|||
|
|
# 5. 验证安装
|
|||
|
|
console.print("🔍 验证PyTorch安装...")
|
|||
|
|
verify_cmd = """python -c "
|
|||
|
|
import torch
|
|||
|
|
print(f'PyTorch版本: {torch.__version__}')
|
|||
|
|
print(f'CUDA可用: {torch.cuda.is_available()}')
|
|||
|
|
if torch.cuda.is_available():
|
|||
|
|
print(f'CUDA版本: {torch.version.cuda}')
|
|||
|
|
print(f'GPU数量: {torch.cuda.device_count()}')
|
|||
|
|
print(f'当前GPU: {torch.cuda.get_device_name(0)}')
|
|||
|
|
"
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
run_command(verify_cmd, "验证PyTorch安装")
|
|||
|
|
|
|||
|
|
console.print(Panel.fit("🎉 安装完成！现在可以继续运行VideoLingo了", style="bold green"))
|
|||
|
|
console.print("💡 提示: 如果还有其他包需要安装，它们将自动使用清华镜像源", style="dim")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|
|||
|
|
|