版本查询:
python 版本:直接 which python 或者 python 进入界面查看;
pytorch 版本:进入 python,import torch, torch.version
cudatoolkit 版本(这里特指 conda 安装的那个 cuda):进入 python,import torch, torch.version.cuda
torchvision 版本:conda list torchvision
系统 cuda:一般在 / user/local/cuda
查看显卡版本:
ubuntu-drivers devices
nvidia-smi
还有一个 cudnn?
NVIDIA cuDNN 是用于深度神经网络的 GPU 加速库。
对应关系:
- 这里 pytorch 和 cudatoolkit 版本对应关系:
https://pytorch.org/get-started/previous-versions/
2.cudatoolkit 版本和系统 cuda 对应关系:
https://www.zhihu.com/question/344950161/answer/818139888
- 系统 cuda 和 nvidia 对应关系:
https://blog.csdn.net/He_9520/article/details/100032803
4.cuda 和 cuDNN 的关系和对应关系:
https://www.jianshu.com/p/622f47f94784
https://www.cnblogs.com/yeran/p/11345990.html
小结:
gpu 有驱动,系统 cuda 用于加速,cudnn 是系统 cuda 的深度学习小应用,cudatoolkit 是深度学习框架,对系统 cuda 的适应器(cudatoolkit 是不完整系统 cuda,可能深度学习只需要用一小部分的系统 cuda?)
其他: