Skip to content

GitLab

Commit 39c58b22 authored by comfyanonymous's avatar comfyanonymous
Browse files

Disable cuda malloc on GTX 750 Ti.

parent d5c0765f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
cuda_malloc.py
View file @ 39c58b22
...@@ -37,7 +37,7 @@ def get_gpu_names(): ...@@ -37,7 +37,7 @@ def get_gpu_names():
return set() return set()
def cuda_malloc_supported(): def cuda_malloc_supported():
blacklist = {"GeForce GTX 960M", "GeForce GTX 950M", "GeForce 945M", "GeForce 940M", "GeForce 930M", "GeForce 920M", "GeForce 910M"} blacklist = {"GeForce GTX 960M", "GeForce GTX 950M", "GeForce 945M", "GeForce 940M", "GeForce 930M", "GeForce 920M", "GeForce 910M", "GeForce GTX 750", "GeForce GTX 745"}
try: try:
names = get_gpu_names() names = get_gpu_names()
except: except:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment