【地表最强】深度学习环境配置攻略 | 【nvidia-driver】, 【cuda toolkit】, 【cudnn】, 【pytorch】

发布时间 2023-10-16 17:14:56作者: zz子木zz

更新截止到 2023.10.16

1.要素:

  1. linux(ubuntu 22.04)
  2. nvidia-driver(也叫做 cuda driver):英伟达GPU驱动,命令:nvidia-smi
  3. cuda (也叫做 cuda toolkit): 这个必须有。CUDA是NVIDIA创建的一个并行计算平台和编程模型(用这个来跑深度学习),命令: nvcc -V
  4. cudnn: nvidia开发的深度学习算子库,用于加速训练(新手阶段不装也行,程序照样能跑)
  5. 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型号和版本

pytorch 和 linux与cuda(nvcc -V)对应表(提供下载)


3.安装线路

  • 线路1:1->3->4->5 (推荐)
    解释一下,因为装cuda的时候会自带一个是否安装对应的最新版本的driver的选项,选yes,就相当于完成了nvidia-driver的安装。
    为什么这样更好?

    1. 省了一步
    2. 不会出现 nvcc -Vnvidia-smi 里的cuda版本不一致的问题
  • 线路2: 1->2->3->4->5
    这样的安装线路,nvidia-smi和nvcc -V的cuda版本可能不一致,但只要牢记pytorch要用的是 nvcc -V里面的pytorch版本即可
    附一个我自己的安装过程记录(线路2)