查看CUDA版本号这里有三种方法。
nvidia-smi
运行nvidia-smi命令:
右上角可以看到CUDA版本号。
CUDA Toolkit
下载地址:https://developer.nvidia.com/cuda-downloads
$nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Fri_Nov__3_17:51:05_Pacific_Daylight_Time_2023
Cuda compilation tools, release 12.3, V12.3.103
Build cuda_12.3.r12.3/compiler.33492891_0
Windows界面:NVIDIA 控制面板
打开NVIDIA 控制面板
然后在组件的NVCUDA64.DLL里能看到CUDA版本。
PyTorch查看
import torch
print("torch.__version__", torch.__version__)
print("torch.version.cuda ", torch.version.cuda)
print("torch.cuda.is_available() ", torch.cuda.is_available())
torch.__version__ 2.1.1+cu121
torch.version.cuda 12.1
torch.cuda.is_available() True