VideoLingo
VideoLingo copied to clipboard
苹果芯片使用videotoolbox硬件加速FFmpeg视频合成
def check_gpu_available(): """检查是否可以使用GPU加速""" system = platform.system() machine = platform.machine()
# 首先检查是否是 Apple Silicon
if system == 'Darwin' and machine == 'arm64':
return 'videotoolbox'
# 然后检查 NVIDIA GPU
try:
if system == 'Windows':
subprocess.run(['nvidia-smi'], capture_output=True, check=True)
return 'nvidia'
elif system == 'Linux':
subprocess.run(['which', 'nvidia-smi'], capture_output=True, check=True)
return 'nvidia'
except subprocess.CalledProcessError:
pass
return None
加速前
佬,请问下是修改这段代码吗 `def check_gpu_available(): #try: #result = subprocess.run(['ffmpeg', '-encoders'], capture_output=True, text=True) #return 'h264_nvenc' in result.stdout #except: #return False """检查是否可以使用GPU加速""" system = platform.system() machine = platform.machine() # 首先检查是否是 Apple Silicon if system == 'Darwin' and machine == 'arm64': return 'videotoolbox'
# 然后检查 NVIDIA GPU
try:
if system == 'Windows':
subprocess.run(['nvidia-smi'], capture_output=True, check=True)
return 'nvidia'
elif system == 'Linux':
subprocess.run(['which', 'nvidia-smi'], capture_output=True, check=True)
return 'nvidia'
except subprocess.CalledProcessError:
pass
return None
`
快是快了,视频文件大的离谱
快是快了,视频文件大的离谱
大概比默认配置大多少?