yuhongtian17 / Spatial-Transform-Decoupling

MIT License
89 stars 7 forks source link

RuntimeError: CUDA error: no kernel image is available for execution on the device #14

Closed liutianyi00 closed 5 months ago

liutianyi00 commented 5 months ago

显卡型号、"nvidia-smi" cuda版本、"nvcc -V" cuda版本和PyTorch版本是否匹配 GPU3090, nvidia-smi:CUDA Version: 12.4, nvcc-V:nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2024 NVIDIA Corporation Built on Tue_Feb_27_16:19:38_PST_2024 Cuda compilation tools, release 12.4, V12.4.99 Build cuda_12.4.r12.4/compiler.33961263_0 pytorch:python=3.7 pytorch==1.7.0 cudatoolkit=10.1 torchvision -c pytorch