总体概念
Gpu作为硬件需要驱动,cuda用于加速,cudnn是系统cuda的深度学习小应用,cudatoolkit是pytorch深度学习框架,对系统cuda的适应器
查看版本命令
硬件能否识别
lspci | grep NVIDIA
lspci | grep VGA
gpu驱动命令:cat /proc/driver/nvidia/version
cuda 版本命令 : cat /usr/local/cuda/version.txt
cudnn 版本命令: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
pytorh 版本:
import torch
torch.__version__
torch.cuda.is_available()
版本匹配
gpu 驱动支持 cuda版本
查看网址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions
cuda支持 pytorch 版本
安装
驱动安装
下载地址:NVidia 显卡 驱动程序
安装说明:https://blog.youkuaiyun.com/u014797226/article/details/79626693
cuda安装
下载地址:CUDA Toolkit Archive | NVIDIA Developer
https://blog.youkuaiyun.com/lihe4151021/article/details/90237681
https://blog.youkuaiyun.com/QLULIBIN/article/details/78714596
https://blog.youkuaiyun.com/wf19930209/article/details/81879514
https://blog.youkuaiyun.com/qq_45049586/article/details/104593173
Cudnn安装
下载地址:https://developer.nvidia.com/rdp/cudnn-archive