BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard

Error building extensions

Open lightandshadow68 opened this issue 6 months ago • 1 comments

I'm attempting to install BasicSR via pip and conda.

After setting the environment variable to enable extension compilation, when I build I receive the following error.

 The detected CUDA version (11.5) mismatches the version that was used to compile
  PyTorch (12.1). Please make sure to use the same CUDA versions.

However, when I check the versions of the drivers and CUDA installed, I receive....

(ml) ubuntu@ip-10-0-83-223:~$ nvidia-smi
Thu Aug 22 18:07:23 2024       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.183.01             Driver Version: 535.183.01   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  Tesla T4                       Off | 00000000:00:1E.0 Off |                    0 |
| N/A   26C    P8               9W /  70W |      2MiB / 15360MiB |      0%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|  No running processes found                                                           |
+---------------------------------------------------------------------------------------+
(ml) ubuntu@ip-10-0-83-223:~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Thu_Nov_18_09:45:30_PST_2021
Cuda compilation tools, release 11.5, V11.5.119
Build cuda_11.5.r11.5/compiler.30672275_0

What's odd is that nvcc and Nvidia-smi do not seem to agree on the version of CUDA installed, or it's referring to the toolkit version, which is different than the actual CUDA api?

lightandshadow68 avatar Aug 22 '24 18:08 lightandshadow68