Spaces:
Build error
Build error
File size: 1,342 Bytes
d5f197c aed8ba9 d5f197c aed8ba9 d5f197c aed8ba9 d5f197c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | # # environment_setup.py
# import subprocess
# import sys
# import os
# def install_packages():
# packages = [
# "einops==0.7.0",
# "protobuf==5.26.1",
# "llama-index==0.11.20",
# "llama-index-llms-replicate==0.3.0",
# "llama-index-llms-openai-like==0.2.0",
# "llama-index-embeddings-huggingface==0.3.1",
# "llama-index-embeddings-instructor==0.2.1",
# "torch==2.5.0",
# "torchvision==0.20.0",
# "torchaudio==2.5.0"
# ]
# index_url = "https://download.pytorch.org/whl/cu121" # 根据你的 CUDA 版本调整
# for package in packages:
# try:
# if package.startswith("torch") or package.startswith("torchvision") or package.startswith("torchaudio"):
# subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}", "--index-url", index_url, "-q"])
# else:
# subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}", "-q"])
# print(f"Successfully installed {package}")
# except subprocess.CalledProcessError as e:
# print(f"Error installing {package}: {e}")
# if __name__ == "__main__":
# print("Starting to install required packages...")
# install_packages()
# print("Package installation completed.")
|