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

2023-12-13 16:59:35

查看CUDA版本号这里有三种方法。

nvidia-smi

运行nvidia-smi命令：
1
右上角可以看到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 控制面板
1

然后在组件的NVCUDA64.DLL里能看到CUDA版本。
1

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

1

文章来源:https://blog.csdn.net/lilongsy/article/details/134898122
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。