查看NVIDIA CUDA版本号的四种方法

查看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