Linux查看CUDA和cuDNN版本
查看CUDA版本
方法一:

nvcc -version
或者

nvcc -V
方法二:

去安装目录下查看:

cat /usr/local/cuda/version.txt

查看cuDNN版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如何没有,则可能没有安装cuDNN

使用 PyTorch 查看 CUDA 和 cuDNN 版本
import torch
print(torch.version)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

只想做个笔记,百度上嫖的

gedit /home/chongge/.conda/envs/cuda_test/etc/conda/activate.d/activate.sh

Logo

更多推荐