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

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 #15

Closed liutianyi00 closed 5 months ago

liutianyi00 commented 5 months ago

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

yuhongtian17 commented 5 months ago

Refer to this

liutianyi00 commented 5 months ago

我的CUDA是12.4，请问应该安装哪个版本的python和pytorch呢