Skip to content

GitLab

Commit 85a8900a authored by comfyanonymous's avatar comfyanonymous
Browse files

Disable cuda malloc on regular GTX 960.

parent 78e7958d
Hide 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