Update infer-web.py
This commit is contained in:
parent
19cc9062b0
commit
28948f8961
@ -23,6 +23,7 @@ from i18n import I18nAuto
|
|||||||
import ffmpeg
|
import ffmpeg
|
||||||
|
|
||||||
i18n = I18nAuto()
|
i18n = I18nAuto()
|
||||||
|
i18n.print()
|
||||||
# 判断是否有能用来训练和加速推理的N卡
|
# 判断是否有能用来训练和加速推理的N卡
|
||||||
ngpu = torch.cuda.device_count()
|
ngpu = torch.cuda.device_count()
|
||||||
gpu_infos = []
|
gpu_infos = []
|
||||||
|
Loading…
Reference in New Issue
Block a user