Linux
查看CUDA版本:
nvcc --version
或
nvcc --V
Windows
查看CUDA版本:
nvcc --version
或进入 CUDA 的安装目录查看:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
使用PyTorch 查看 CUDA 版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())