apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2024-08-24 08:31:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bruh moment 11

Browse Source
This commit is contained in:
Somdev Sangwan 2023-05-31 10:38:16 +05:30 committed by GitHub
parent 6a09ee0c4b
commit 7e68260f71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run.py
View File

@ -68,7 +68,7 @@ def pre_check():
CUDA_VERSION = torch.version.cuda CUDA_VERSION = torch.version.cuda
CUDNN_VERSION = torch.backends.cudnn.version() CUDNN_VERSION = torch.backends.cudnn.version()
NVIDIA_PROVIDERS = ['CUDAExecutionProvider', 'TensorrtExecutionProvider'] NVIDIA_PROVIDERS = ['CUDAExecutionProvider', 'TensorrtExecutionProvider']
if list(set(core.globals.providers) - set(NVIDIA_PROVIDERS)) > 1: if len(list(set(core.globals.providers) - set(NVIDIA_PROVIDERS))) > 1:
if not torch.cuda.is_available() or not CUDA_VERSION: if not torch.cuda.is_available() or not CUDA_VERSION:
quit("You are using --gpu flag but CUDA isn't available or properly installed on your system.") quit("You are using --gpu flag but CUDA isn't available or properly installed on your system.")
if CUDA_VERSION > '11.8': if CUDA_VERSION > '11.8':