Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 85a8900a authored by comfyanonymous's avatar comfyanonymous
Browse files

Disable cuda malloc on regular GTX 960.

parent 78e7958d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
cuda_malloc.py
View file @ 85a8900a
...@@ -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", "GeForce GTX 750", "GeForce GTX 745"} blacklist = {"GeForce GTX 960", "GeForce GTX 950", "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