一、查看CUDA版本
方法一:
nvcc -version
或者
nvcc -V
方法二:
cat /usr/local/cuda/version.txt
二、查看cuDNN版本
使用 PyTorch 查看 CUDA 和 cuDNN 版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
一、查看CUDA版本
方法一:
nvcc -version
或者
nvcc -V
方法二:
cat /usr/local/cuda/version.txt
二、查看cuDNN版本
使用 PyTorch 查看 CUDA 和 cuDNN 版本
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())