Ma/video-subtitle-remover
1
0
Fork 0
mirror of https://github.com/YaoFANGUK/video-subtitle-remover.git synced 2026-06-23 05:23:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Ma:main
Branches Tags
Ma:main Ma:packaging-windows Ma:packaging-macos
Ma:1.4.0 Ma:1.1.1 Ma:1.1.0 Ma:1.0.0
...
compare: Ma:d2695a12621ea1c38f29c64fafbd264160c4d20e
Branches Tags
Ma:main Ma:packaging-windows Ma:packaging-macos
Ma:1.4.0 Ma:1.1.1 Ma:1.1.0 Ma:1.0.0

2 Commits
main ... d2695a1262

Author SHA1 Message Date
年轻而且有才
d2695a1262 Merge 5ac9f67ad5 into 196a1a8e7b 2025-04-25 11:11:20 +08:00
xshaitt
5ac9f67ad5 优化 get_infer_gpuid 函数，增加对 PaddlePaddle 版本的检查 2025-04-18 11:22:02 +08:00
1 changed files with 8 additions and 1 deletions
Expand all files Collapse all files

9
backend/tools/infer/utility.py
View File

@@ -340,7 +340,14 @@ def get_infer_gpuid():
if sysstr == "Windows":
return 0
if not paddle.fluid.core.is_compiled_with_rocm():
# 检查 PaddlePaddle 版本
paddle_version = paddle.__version__
major_version = int(paddle_version.split('.')[0])
if major_version >= 2:
is_rocm = paddle.device.is_compiled_with_rocm()
else:
is_rocm = paddle.fluid.core.is_compiled_with_rocm()
if not is_rocm:
cmd = "env | grep CUDA_VISIBLE_DEVICES"
else:
cmd = "env | grep HIP_VISIBLE_DEVICES"