-
Notifications
You must be signed in to change notification settings - Fork 2k
Description
我的conda环境如下:
GPU 架构: (12, 0)
CUDA: 12.6
PyTorch: 2.7.0+cu126
Python: 3.10.15 | packaged by conda-forge | (main, Oct 16 2024, 01:15:49) [MSC v.1941 64 bit (AMD64)]
torchaudio2.7.0+cu126
CUDA可用: True
GPU设备数: 1
IDE定位到model.py文件350行代码:
flow_encoder = torch.jit.load(flow_encoder_model, map_location=self.device)
并发生异常: RuntimeError
CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.
控制台信息:
UserWarning:
NVIDIA GeForce RTX 5060 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90.
If you want to use the NVIDIA GeForce RTX 5060 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/