Install bug
The previous steps did not report any errors until running python test.py.
(vmt2) jyw@star-ESC8000-G4:~/newvmt/vmt/models/ops$ python test.py
Traceback (most recent call last):
File "test.py", line 24, in
extern "C" launch_bounds(512, 4) global void reduction_prod_kernel(ReduceJitOp r){ r.run(); } nvrtc: error: invalid value for --gpu-architecture (-arch)。
(vmt2) jyw@star-ESC8000-G4:~/newvmt/vmt/models/ops$ nvidia-smi
Fri Nov 1 17:16:27 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 560.35.03 Driver Version: 560.35.03 CUDA Version: 12.6 |
|-----------------------------------------+------------------------+----------------------+
| 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 NVIDIA RTX 6000 Ada Gene... Off | 00000000:20:00.0 On | Off |
| 30% 36C P8 21W / 300W | 2MiB / 49140MiB | 0% Default |
| | | N/A |
+-----------------------------------------+------------------------+----------------------+
| 1 NVIDIA RTX 6000 Ada Gene... Off | 00000000:23:00.0 Off | Off |
| 30% 36C P8 22W / 300W | 2MiB / 49140MiB | 0% Default |
| | | N/A |
+-----------------------------------------+------------------------+----------------------+
+-----------------------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=========================================================================================| | No running processes found | +-----------------------------------------------------------------------------------------+ (vmt2) jyw@star-ESC8000-G4:~/newvmt/vmt/models/ops$ nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Sun_Mar_21_19:15:46_PDT_2021 Cuda compilation tools, release 11.3, V11.3.58 Build cuda_11.3.r11.3/compiler.29745058_0 (vmt2) jyw@star-ESC8000-G4:~/newvmt/vmt/models/ops$ python Python 3.8.20 (default, Oct 3 2024, 15:24:27) [GCC 11.2.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.
import torch torch.version '1.12.1+cu113' torch.cuda.is_available() True