diff --git a/desktop_env/envs/__init__.py b/desktop_env/envs/__init__.py index 3f54d12..3dd509c 100644 --- a/desktop_env/envs/__init__.py +++ b/desktop_env/envs/__init__.py @@ -189,9 +189,12 @@ def _install_virtual_machine(vm_name, working_dir="./vm_data", downloaded_file_n os.makedirs(working_dir, exist_ok=True) def __download_and_unzip_vm(): # Determine the platform and CPU architecture to decide the correct VM image to download - if platform.machine() == 'arm64': # macOS with Apple Silicon - url = UBUNTU_ARM_URL - elif platform.machine().lower() in ['amd64', "x86_64"]: + if platform.system() == 'Darwin': # macOS + if os.uname().machine == 'arm64': # Apple Silicon + url = UBUNTU_ARM_URL + else: + url = UBUNTU_X86_URL + elif platform.machine().lower() in ['amd64', 'x86_64']: url = UBUNTU_X86_URL else: raise Exception("Unsupported platform or architecture")