WangJiuming / YOLOOP

MIT License
1 stars 0 forks source link

Some question of YOLOOP's CUDA and Torch #4

Closed qqyanyang closed 2 weeks ago

qqyanyang commented 2 weeks ago

Hi, I am reproducing your program. After installing CUDA12.1 and the corresponding cudnn 8.9.6, and configuring the corresponding environment, I found out during execution that I needed the cuda version 11.7 in order to recognize the GPU. What should I do now? Here's what I found when I ran the check. Could you please help me out?

import torch torch.cuda.is_available() False torch.version.cuda '11.7'

$ nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2023 NVIDIA Corporation Built on Tue_Feb__7_19:32:13_PST_2023 Cuda compilation tools, release 12.1, V12.1.66 Build cuda_12.1.r12.1/compiler.32415258_0