clore
/
hosting
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Nvidia driver 550 install flag | auto pull selftest #2

Merged
clore merged 5 commits from xfs into main 2024-12-11 00:24:01 +00:00
Conversation 0 Commits 5 Files Changed 5 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 8b9f89ff64 - Show all commits

2
lib/clore_partner.py
View File

@ -257,7 +257,7 @@ async def check_to_pull_selftest(current_specs):
if nvidia_gpu["pcie_width"] < min_width: if nvidia_gpu["pcie_width"] < min_width:
min_width = nvidia_gpu["pcie_width"] min_width = nvidia_gpu["pcie_width"]
if " MiB" in nvidia_gpu["mem_total"]: if " MiB" in nvidia_gpu["mem_total"]:
gpu_total_vram += int(nvidia_gpu["mem_total"]) gpu_total_vram += int(nvidia_gpu["mem_total"].replace(" MiB", ''))
if gpu_name in auto_pull_selftest_gpus and current_specs["ram"] > 7 and int(current_specs["cpus"].split('/')[0]) >= 4 and not mixed_cards and min_width > 1 and gpu_total_vram < current_specs["ram"] * 1024 and float(current_specs["disk"].split(' ')[-1].replace("GB", '')) > 25: if gpu_name in auto_pull_selftest_gpus and current_specs["ram"] > 7 and int(current_specs["cpus"].split('/')[0]) >= 4 and not mixed_cards and min_width > 1 and gpu_total_vram < current_specs["ram"] * 1024 and float(current_specs["disk"].split(' ')[-1].replace("GB", '')) > 25:
await utils.async_run_command("docker pull vastai/test:selftest", 14400, non_interactive_env) await utils.async_run_command("docker pull vastai/test:selftest", 14400, non_interactive_env)
except Exception as e: except Exception as e: