Spaces:
Running
Running
| # gpuchecker.py | |
| import re | |
| from fastapi import APIRouter | |
| from fastapi.responses import PlainTextResponse | |
| router = APIRouter() | |
| def _check_graphics_card_info(supported_models, unsupported_models, graphics_card_info): | |
| for model_pattern, descr in supported_models.items(): | |
| if match := re.search(model_pattern, graphics_card_info, re.I): | |
| return True, match.group(), descr.format(*match.groups()) | |
| for model_pattern, descr in unsupported_models.items(): | |
| if match := re.search(model_pattern, graphics_card_info, re.I): | |
| return False, match.group(), descr.format(*match.groups()) | |
| return False, None, None | |
| def _check_amd(graphics_card_info): | |
| supported_models = { | |
| r"Radeon\s*6\d{2}([A-Z])?\b": "this model belongs to the RDNA 2 architecture", | |
| r"(Radeon\s*)?R9\s*[A-Z0-9]+": "R9 models belongs to the GCN 1st gen or newer architecture", | |
| r"(Radeon\s*)?(Pro\s*)?\bW7\d{3}(X)?\b(\s*Duo)?": "Radeon Pro W7000 models belongs to the RDNA 3 architecture", | |
| r"(Radeon\s*)?(Pro\s*)?\bW([5-6])\d{3}(X)?\b(\s*Duo)?": "Radeon Pro W{2}000 models belongs to the RDNA 2 architecture", | |
| r"(AMD\s*)?6800 XT": "this model belongs to the RDNA 2 architecture", | |
| r"Radeon\s*(\(TM\)\s*)?RX Vega(\s*\d{2}\b)": "Radeon RX Vega models belongs to the GCN 5th gen architecture", | |
| r"Radeon Pro Vega ((\d{2}(X)?|II)\b)?(\s*Duo)?": "Radeon Pro Vega models belongs to the GCN 5th gen architecture", | |
| r"Radeon\s*(\(TM\)\s*)?Pro [4-5]\d{2}(X)?": "Radeon Pro 400/500 series belongs to the GCN 4th gen architecture", | |
| r"Radeon VII": "RX models belongs to the GCN 5 architecture", | |
| r"Radeon Graphics \(renoir": "this model belongs to the GCN 5th gen architecture (Vega)", | |
| r"Radeon\s*(\(TM\)\s*)?Vega 8 (Graphics )?\(raven[^)]+\)": "this model belongs to the GCN 5th gen architecture (Vega)", | |
| r"Radeon\s*(\(TM\)\s*)?(Pro\s*)?WX\s*(5\d{3}\b)": "this model belongs to the GCN 4th gen architecture", | |
| r"FirePro": "FirePro models belongs to the GCN 1st gen or newer architecture", | |
| r"HD\s*(7[7-9]\d{2})": "HD {0} model belongs to the GCN 1st gen or newer architecture", | |
| r"(Radeon\s*)?RX\s*([5-7]\d{3})(M|X)?(\s*(XT|Series|S|XTX))?\b": "RX models belongs to the GCN 1st gen or newer architecture", | |
| r"(Radeon\s*)?(RX\s*)?6(3|4)0\b": "it has Polaris 23 chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?62(0|5)\b": "it has Polaris 24 chip that belongs to GCN 3st gen architecture", | |
| r"(Radeon\s*)?610\b": "it has Banks chip that belongs to GCN 1st gen architecture", | |
| r"(Radeon\s*)?RX\s*580(X?)\b": "it has Polaris 20 XT chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?RX\s*570\b": "it has Ellesmere Pro chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?RX\s*560X\b": "it has Polaris 31 XL chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?RX\s*560\b": "it has Baffin XT chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?5(40X|50X)\b": "it has Polaris 23 XT chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?RX\s*5(40|50)\b": "it has Lexa Pro chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?RX\s*480\b": "it has Arctic Islands chip that belongs to GCN 4th gen architecture", | |
| r"(Radeon\s*)?(\(TM\)\s*)?RX\s*4[6-8]0(\b|D)": "it has Ellesmere chip that belongs to GCN 4st gen architecture", | |
| r"(Radeon\s*)?5(30X|35)\b": "it has Polaris 24 XT chip that belongs to GCN 3rd gen architecture", | |
| r"(Radeon\s*)?530\b": "it has Weston chip that belongs to GCN 3rd gen architecture", | |
| r"(Radeon\s*)?520\b": "it has Banks chip that belongs to GCN 1st gen architecture", | |
| r"(Radeon\s*)?(\(TM\)\s*)?R4": "Radeon R4 models belongs to the GCN 1st gen or newer architecture", | |
| r"(Radeon\s*)?(\(TM\)\s*)?R5 (M)?335": "Radeon R5 M335 belongs to the GCN 1st gen architecture", | |
| r"(Radeon\s*)?(\(TM\)\s*)?R7 (M)?2\d{2}(E|X)?\b": "Radeon R7 200 models belongs to GCN 1st or 2nd gen architecture", | |
| r"(Radeon\s*)?(\(TM\)\s*)?R5 (M)?24\d(E|X)?\b": "Radeon R5 240 models belongs to GCN 1st gen architecture", | |
| # r"Radeon\s*(\(TM\)\s*)?(Pro\s*)?Vega (Pro\s*)?": "this model belongs to the GCN 4th gen architecture", | |
| # Add more model-to-architecture mappings as needed | |
| } | |
| unsupported_models = { | |
| r"HD ([5-6])\d{3}": "HD {0}XXX models have TeraScale architecture that is older than GCN 1st gen", | |
| r"HD\s*(7[3-6]\d{2})": "HD {0} model has TeraScale 2 architecture that is older than GCN 1st gen", | |
| r"Radeon R5 (M)?2(2|3)\d(X)?\b": "Radeon R5 220/230 models belongs to Terascale 2 architecture that is older than GCN 1st gen", | |
| r"(AMD\s*ATI\s*)?Radeon\s*680M": "AMD ATI Radeon 680M has TeraScale architecture that is older than GCN 1st gen", | |
| # Add more model-to-architecture mappings as needed | |
| } | |
| return _check_graphics_card_info(supported_models, unsupported_models, graphics_card_info) | |
| def _check_nvidia(graphics_card_info): | |
| supported_models = { | |
| r"(GeForce )?(RTX\s*)?(?<!\d)([2-4])0[5-9]\d(\s*(RTX|Ti))?\b": "RTX {2}0 series are newer than GTX 400", | |
| r"(GeForce )?(GTX\s*)?(?<!\d)16[5-9]\d(\s*(GTX|Ti))?\b": "GTX 16 series are newer than GTX 400", | |
| r"(GeForce )?(GTX\s*)?(?<!\d)10[5-9]\d(\s*(GTX|Ti))?\b": "GTX 10 series are newer than GTX 400", | |
| r"(GTX )?TITAN": "GTX TITAN models are newer than GTX 400", | |
| r"(RTX )?\bA(\d+)": "RTX A models are newer than GTX 400", | |
| r"Quadro FX \d+": "Quadro FX series uses a Quadro-based architecture", | |
| r"Quadro RTX \d+": "Quadro RTX series uses a Quadro-based architecture", | |
| r"Quadro (K|M|P|GP|GV)?\d+(M)?": "it uses a Quadro-based architecture", | |
| r"NVS 8\d{2}(s)?\b": "it uses a Maxwell based architecture", | |
| r"(Quadro )?NVS 110M\b": "it uses a Maxwell based architecture", | |
| r"(GeForce )?GT 730\b": "GeForce from 700 series are newer than GTX 400. It also has 2 or 4 GB", | |
| r"(GeForce )?GTX ([4-9])\d{2}(\s*(GTX|Ti))?\b": "GPUs from GTX {1}00 series are newer than GTX 400", | |
| r"(GeForce )?\bMX\d{3}\b": "MX models are newer than GTX 400", | |
| r"Tesla (.+)": "it has a Tesla architecture", | |
| # Add more model-to-architecture mappings as needed | |
| } | |
| unsupported_models = { | |
| r"(GeForce )(GTX )?3\d{2}": "GTX 3XX models are older than GeForce 400", | |
| r"(Quadro )?NVS 50\b": "although quadro, it only supports opengl 1.3 and is older than 10 years", | |
| r"(Quadro )?NVS \d{3}(s)?\b": "it is older than 10 years", | |
| r"(Quadro )?NVS 1[1-2]0M\b": "it is Curie-based and older than 10 years", | |
| r"(Quadro )?NVS 1\d{2}M\b": "although it is Tesla-based it is older than 10 years", | |
| r"(Quadro )?NVS 4200M\b": "although it has Fermi architecture (newer than Tesla) it is older than 10 years", | |
| # Add unsupported model-to-architecture mappings if needed | |
| } | |
| return _check_graphics_card_info(supported_models, unsupported_models, graphics_card_info) | |
| def _check_intel(graphics_card_info): | |
| supported_models = { | |
| r"HD (Graphics )?5\d{3}\b": "it has the Broadwell architecture and is less than 10 years old (from 2015)", | |
| r"(Mesa\s*)?(Iris\s*)?Xe Graphics": "Tiger Lake is newer than Broadwell architecture", | |
| r"Iris Plus Graphics G7": "Ice Lake is newer than Broadwell architecture", | |
| r"UHD\s*(Graphics )?6[3-5]\d\b": "Coffee Lake or Comet Lake are newer than Broadwell architecture", | |
| r"UHD\s*(Graphics )?62\d\b": "Kaby Lake is newer than Broadwell architecture", | |
| r"HD\s*(Graphics )?(P)?6[1-3]\d\b": "Kaby Lake is newer than Broadwell architecture", | |
| r"UHD\s*(Graphics )?60\d": "Gemini Lake is newer than Broadwell architecture", | |
| r"UHD Graphics": "Kaby Lake, Coffee Lake or Comet Lake are newer than Broadwell architecture", | |
| r"Iris": "Coffee Lake is newer than Broadwell architecture", | |
| r"HD (Graphics )?5\d{2}\b": "Skylake is newer than Broadwell architecture", | |
| r"Iris (Graphics )?6\d{3}\b": "it has the Broadwell architecture", | |
| r"Intel(\(R\))? (Arc(\(TM\))?\s*)?(A)?7\d{2}\b": "the A770 model is based on the Intel Arc architecture that is newer than Broadwell", | |
| r"Intel\s*(Arc\s*)?(A)?7\d{2}\b": "the A770 model is based on the Intel Arc architecture that is newer than Broadwell", | |
| # Add more model-to-architecture mappings as needed | |
| } | |
| unsupported_models = { | |
| r"HD (Graphics )?4\d{3}\b": "it has the Haswell architecture that is older than Broadwell architecture. Also [Due to driver issues, support for Intel HD4000 series GPUs has been dropped](https://wiki.blender.org/wiki/Reference/Release_Notes/4.0#:~:text=Due%20to%20driver%20issues%2C%20support%20for%20Intel%20HD4000%20series%20GPUs%20has%20been%20dropped)", | |
| r"HD Graphics 3\d{3}\b": "Sandy Bridge is older than Broadwell architecture" | |
| # Add unsupported model-to-architecture mappings if needed | |
| } | |
| return _check_graphics_card_info(supported_models, unsupported_models, graphics_card_info) | |
| def _check_apple(graphics_card_info): | |
| supported_models = { | |
| r"(Apple\s*)?(`)?\bM1(`)?(\s*Max)?": "it is one of the new ARM-based system designed by Apple Inc", | |
| r"(Apple\s*)?(`)?\bM2(`)?(\s*Max)?": "it is one of the new ARM-based system designed by Apple Inc", | |
| # Add more model-to-architecture mappings as needed | |
| } | |
| unsupported_models = { | |
| # Add unsupported model-to-architecture mappings if needed | |
| } | |
| return _check_graphics_card_info(supported_models, unsupported_models, graphics_card_info) | |
| def _check_apple_os_version(os_version, is_apple_silicon): | |
| major, minor = map(int, os_version.split(".")[:2]) | |
| if is_apple_silicon: | |
| if major >= 11: | |
| return True | |
| else: | |
| if major >= 10 and minor >= 15: | |
| return True | |
| return False | |
| def gpu_checker_get_message(text): | |
| is_supported = False | |
| vendor = None | |
| model = None | |
| descr = None | |
| if "nvidia" in text.lower() or "rtx" in text.lower() or "gtx" in text.lower() or "geforce" in text.lower(): | |
| vendor = 'NVIDIA' | |
| is_supported, model, descr = _check_nvidia(text) | |
| elif "amd " in text.lower() or "ati " in text.lower() or "radeon" in text.lower(): | |
| vendor = 'AMD' | |
| is_supported, model, descr = _check_amd(text) | |
| elif "intel" in text.lower(): | |
| vendor = 'Intel' | |
| is_supported, model, descr = _check_intel(text) | |
| elif "apple" in text.lower() or re.search(r'\bM1\b', text): | |
| vendor = 'Apple' | |
| is_supported, model, descr = _check_apple(text) | |
| if not vendor: | |
| return "Could not find graphics card information" | |
| elif not model: | |
| return f"Could not determine the card model from {vendor}" | |
| message = f"The {vendor} card {model} is {'supported' if is_supported else 'not supported'} as {descr}" | |
| if not is_supported: | |
| message += """ | |
| This GPU is below the minimum requirements for Blender, so Blender no longer provide support for it. https://www.blender.org/download/requirements/ | |
| Installing the latest graphics driver sometimes helps to make such GPUs work, see here for more information. https://docs.blender.org/manual/en/dev/troubleshooting/gpu/index.html | |
| If that doesn't help, you can use Blender 2.79: https://www.blender.org/download/previous-versions/ | |
| """ | |
| return message | |
| def gpu_checker(gpu_info: str = "", response_class=PlainTextResponse): | |
| message = gpu_checker_get_message(gpu_info) | |
| return message | |
| if __name__ == "__main__": | |
| gpu_info = "AMD Radeon HD 7660D" | |
| message = gpu_checker_get_message(gpu_info) | |
| print(message) | |