已正确安装但无法通过GPU/CUDA运行Whisper AI转录的问题求助
问题:RTX3070笔记本无法通过GPU运行Whisper AI转录,触发HPU模块加载错误
我使用搭载RTX3070显卡的技嘉Aorus笔记本,尝试通过CUDA/GPU运行Whisper AI转录功能时,每次操作都会触发相同错误。若不指定device参数,程序默认使用CPU可正常运行,但指定GPU就会报错。
错误日志
File "C:\Python\lib\runpy.py", line 196, in _run_module_as_main return run_code(code, main_globals, None, File "C:\Python\lib\runpy.py", line 86, in run_code exec(code, run_globals) File "C:\Python\Scripts\whisper.exe_main.py", line 7, in File "C:\Python\lib\site-packages\whisper\transcribe.py", line 444, in cli model = load_model(model_name, device=device, download_root=model_dir) File "C:\Python\lib\site-packages\whisper\__init__.py", line 144, in load_model checkpoint = torch.load(fp, map_location=device) File "C:\Python\lib\site-packages\torch\serialization.py", line 1014, in load return _load(opened_zipfile, File "C:\Python\lib\site-packages\torch\serialization.py", line 1422, in _load result = unpickler.load() File "C:\Python\lib\site-packages\torch\serialization.py", line 1392, in persistent_load typed_storage = load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location)) File "C:\Python\lib\site-packages\torch\serialization.py", line 1366, in load_tensor wrap_storage=restore_location(storage, location), File "C:\Python\lib\site-packages\torch\serialization.py", line 1296, in restore_location return default_restore_location(storage, map_location) File "C:\Python\lib\site-packages\torch\serialization.py", line 381, in default_restore_location result = fn(storage, location) File "C:\Python\lib\site-packages\torch\serialization.py", line 304, in _hpu_deserialize assert hpu is not None, "HPU device module is not loaded" AssertionError: HPU device module is not loaded
已执行的排查操作
- 已安装NVIDIA toolkit,测试
torch.cuda.isAvailable()返回true - 执行过
pip install nvidia-cuda-runtime-cu12安装对应CUDA运行时 - 可通过任务管理器和AORUS控制中心的仪表盘检查GPU使用率
内容的提问来源于stack exchange,提问作者Cristian Vintila
相关产品推荐
相关产品推荐

