更新截止到 2023.10.16
1.要素:
- linux(ubuntu 22.04)
- nvidia-driver(也叫做 cuda driver):英伟达GPU驱动,命令:
nvidia-smi
- cuda (也叫做 cuda toolkit): 这个必须有。CUDA是NVIDIA创建的一个并行计算平台和编程模型(用这个来跑深度学习),命令:
nvcc -V
- cudnn: nvidia开发的深度学习算子库,用于加速训练(新手阶段不装也行,程序照样能跑)
- pytorch: 深度学习框架
2.依赖关系:
显卡 --> driver --> cuda(nvcc -V
) --> cudnn & pytorch
1.显卡型号 --决定--> driver版本
nvidia显卡驱动下载网址
在这里搜索显卡的型号,可以看到能用的显卡驱动版本并可以下载驱动
2.driver版本 --决定--> cuda(nvcc -V
)版本
cuda和driver版本对应表 (不提供下载)
历代版本cuda下载地址
3.cuda版本决定 --> cudnn版本
cuda和cudnn版本对应表
(点击即可下载对应版本的cudnn)
4.linux版本 & cuda(nvcc -V
)版本 --决定--> pytorch型号和版本
3.安装线路
-
线路1:1->3->4->5 (推荐)
解释一下,因为装cuda的时候会自带一个是否安装对应的最新版本的driver的选项,选yes,就相当于完成了nvidia-driver的安装。
为什么这样更好?- 省了一步
- 不会出现
nvcc -V
和nvidia-smi
里的cuda版本不一致的问题
-
线路2: 1->2->3->4->5
这样的安装线路,nvidia-smi和nvcc -V的cuda版本可能不一致,但只要牢记pytorch要用的是nvcc -V
里面的pytorch版本即可
附一个我自己的安装过程记录(线路2)