import os import subprocess import sys import time import platform import re import logging # 获取当前脚本所在目录的父目录（项目根目录） script_dir = os.path.dirname(os.path.abspath(__file__)) project_root = os.path.dirname(script_dir) # 配置日志 - 使用相对路径 log_file = os.path.join(project_root, 'environment_check.log') logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[ logging.StreamHandler(), logging.FileHandler(log_file) ] ) def run_command(command, description=None, show_output=False): """运行命令并打印状态""" if description: logging.info(f"=== {description} ===") logging.info(f"执行: {command}") try: # 明确指定使用bash执行命令 result = subprocess.run(['/bin/bash', '-c', command], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) output = result.stdout.decode('utf-8') logging.info(f"命令执行成功: {command}") if show_output: logging.info(f"输出:\n{output}") return True, output except subprocess.CalledProcessError as e: error = e.stderr.decode('utf-8') logging.error(f"命令执行失败: {command}") logging.error(f"错误信息: {error}") return False, error def check_dependencies(): """检查系统是否满足RoboTwin的依赖要求""" results = [] all_passed = True # 打印依赖要求 logging.info("\n=== RoboTwin Dependencies ===") logging.info(""" Python versions: * Python 3.8, 3.10 Operating systems: * Linux: Ubuntu 18.04+, Centos 7+ Hardware: * Rendering: NVIDIA or AMD GPU * Recommended CUDA Version: 12.1 * Ray tracing: NVIDIA RTX GPU or AMD equivalent * Ray-tracing Denoising: NVIDIA GPU * GPU Simulation: NVIDIA GPU Software: * Ray tracing: NVIDIA Driver >= 470 * Denoising (OIDN): NVIDIA Driver >= 520 * CUDA Version: 12.1 * Conda: Required """) # 检查操作系统 os_check = {"name": "操作系统检查", "passed": False, "details": ""} os_name = platform.system() if os_name != "Linux": os_check["details"] = f"不支持的操作系统: {os_name}，需要 Linux" else: # 检查Linux发行版 success, distro_info = run_command("cat /etc/os-release", show_output=False) if success: # 检查是否是Ubuntu或CentOS if "Ubuntu" in distro_info: # 提取Ubuntu版本 match = re.search(r'VERSION_ID="(\d+\.\d+)"', distro_info) if match and float(match.group(1)) >= 18.04: os_check["passed"] = True os_check["details"] = f"Ubuntu {match.group(1)}，满足要求 (Ubuntu 18.04+)" else: os_check["details"] = "Ubuntu版本过低，需要 Ubuntu 18.04+" elif "CentOS" in distro_info: # 提取CentOS版本 match = re.search(r'VERSION_ID="(\d+)"', distro_info) if match and int(match.group(1)) >= 7: os_check["passed"] = True os_check["details"] = f"CentOS {match.group(1)}，满足要求 (CentOS 7+)" else: os_check["details"] = "CentOS版本过低，需要 CentOS 7+" else: os_check["details"] = "不支持的Linux发行版，需要 Ubuntu 18.04+ 或 CentOS 7+" else: os_check["details"] = "无法确定Linux发行版" results.append(os_check) all_passed = all_passed and os_check["passed"] # 检查Conda是否已安装 conda_check = {"name": "Conda检查", "passed": False, "details": ""} success, conda_info = run_command("conda --version", "检查Conda是否已安装", show_output=False) if success: # 提取Conda版本 match = re.search(r"conda (\d+\.\d+\.\d+)", conda_info) if match: conda_version = match.group(1) conda_check["passed"] = True conda_check["details"] = f"Conda已安装 (版本 {conda_version})" # 检查conda初始化状态 success, conda_init = run_command("conda info | grep -i 'initialized'", "检查Conda初始化状态", show_output=False) if success and "yes" in conda_init.lower(): conda_check["details"] += ", 已初始化" else: conda_check["details"] += ", 但可能未正确初始化，请运行 'conda init bash'" else: conda_check["details"] = "无法确定Conda版本，但似乎已安装" else: conda_check["details"] = "未安装Conda，请先安装Conda" results.append(conda_check) all_passed = all_passed and conda_check["passed"] # 检查NVIDIA GPU gpu_check = {"name": "GPU检查", "passed": False, "details": ""} success, gpu_info = run_command("nvidia-smi", "检查NVIDIA GPU", show_output=False) if success: # 检查是否检测到NVIDIA GPU if "NVIDIA" in gpu_info: gpu_check["passed"] = True gpu_check["details"] = "检测到NVIDIA GPU" else: gpu_check["details"] = "未检测到NVIDIA GPU" else: # 尝试检查AMD GPU success, amd_gpu_info = run_command("lspci | grep -i amd", "检查AMD GPU", show_output=False) if success and "AMD" in amd_gpu_info and ("Graphics" in amd_gpu_info or "VGA" in amd_gpu_info): gpu_check["passed"] = True gpu_check["details"] = "检测到AMD GPU" else: gpu_check["details"] = "未检测到NVIDIA或AMD GPU，需要NVIDIA或AMD GPU" results.append(gpu_check) all_passed = all_passed and gpu_check["passed"] # 检查NVIDIA驱动版本 driver_check = {"name": "NVIDIA驱动版本检查", "passed": False, "details": ""} if success and "NVIDIA" in gpu_info: match = re.search(r"Driver Version: (\d+\.\d+)", gpu_info) if match: driver_version = float(match.group(1)) if driver_version >= 470: driver_check["passed"] = True driver_check["details"] = f"NVIDIA驱动版本 {driver_version}，满足要求 (>=470)" # 检查是否满足光线追踪去噪要求 if driver_version >= 520: driver_check["details"] += "，同时满足去噪要求 (>=520)" else: driver_check["details"] += "，但不满足去噪要求 (>=520)" else: driver_check["details"] = f"NVIDIA驱动版本 {driver_version} 过低，需要 >=470" else: driver_check["details"] = "无法确定NVIDIA驱动版本" elif gpu_check["passed"]: driver_check["passed"] = True driver_check["details"] = "使用AMD GPU，跳过NVIDIA驱动版本检查" else: driver_check["details"] = "未检测到NVIDIA GPU，无法检查驱动版本" results.append(driver_check) all_passed = all_passed and driver_check["passed"] # 检查CUDA版本 cuda_check = {"name": "CUDA版本检查", "passed": False, "details": ""} if success and "NVIDIA" in gpu_info: # 尝试获取CUDA版本 success, cuda_info = run_command("nvcc --version", "检查CUDA版本", show_output=False) if success: match = re.search(r"release (\d+\.\d+)", cuda_info) if match: cuda_version = float(match.group(1)) if abs(cuda_version - 12.1) < 0.1: # 允许小的版本差异，比如12.1和12.0 cuda_check["passed"] = True cuda_check["details"] = f"CUDA版本 {cuda_version}，满足建议版本 (12.1)" else: cuda_check["passed"] = True # 仍然通过，但发出警告 cuda_check["details"] = f"CUDA版本 {cuda_version}，不是推荐的 12.1，可能会遇到兼容性问题" else: cuda_check["details"] = "无法确定CUDA版本" else: # 尝试使用nvidia-smi确定CUDA版本 match = re.search(r"CUDA Version: (\d+\.\d+)", gpu_info) if match: cuda_version = float(match.group(1)) if abs(cuda_version - 12.1) < 0.1: cuda_check["passed"] = True cuda_check["details"] = f"CUDA版本 {cuda_version}，满足建议版本 (12.1)" else: cuda_check["passed"] = True # 仍然通过，但发出警告 cuda_check["details"] = f"CUDA版本 {cuda_version}，不是推荐的 12.1，可能会遇到兼容性问题" else: cuda_check["details"] = "未安装CUDA或无法确定版本" elif gpu_check["passed"]: cuda_check["passed"] = True cuda_check["details"] = "使用AMD GPU，跳过CUDA版本检查" else: cuda_check["details"] = "未检测到NVIDIA GPU，无法检查CUDA版本" results.append(cuda_check) all_passed = all_passed and cuda_check["passed"] # 打印检查结果 logging.info("\n=== 环境依赖检查结果 ===") for result in results: status = "✓" if result["passed"] else "✗" logging.info(f"{status} {result['name']}: {result['details']}") if not all_passed: logging.warning("\n警告：某些依赖项不满足要求，环境配置可能无法正常完成或运行。") return all_passed, results if __name__ == "__main__": all_passed, results = check_dependencies() # 将结果保存到项目根目录 import json results_file = os.path.join(project_root, 'environment_check_results.json') with open(results_file, 'w') as f: json.dump({"dependencies_met": all_passed, "check_results": results}, f) # 退出状态码 if all_passed: sys.exit(0) else: sys.exit(1)