版本查詢:
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加速庫。
對應關系:
1.這里pytorch和cudatoolkit版本對應關系:
https://pytorch.org/get-started/previous-versions/
2.cudatoolkit版本和系統cuda對應關系:
https://www.zhihu.com/question/344950161/answer/818139888
3.系統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?)
其他: