深度学习GPU的环境太乱?从头配置环境:理解+实操

深度学习CUDA环境太乱?环境配置:理解+实操 小白关于环境配置的一点理解 0.装环境之前你需要知道有关CUDA的事 进一步理解(optional) 如何查看你当前的GPU型号? 1. 装CUDA驱动(即CUDA toolkit) 1.1

深度学习CUDA环境太乱?环境配置:理解+实操

  • 小白关于环境配置的一点理解
  • 0.装环境之前你需要知道有关CUDA的事
    • 进一步理解(optional)
    • 如何查看你当前的GPU型号?
  • 1. 装CUDA驱动(即CUDA toolkit)
    • 1.1
    • 1.2
    • 1.3
    • 注:
    • 1.4
  • 2.安装cuDNN
    • 2.1
    • 2.2
    • 2.3
  • 3. 装对应的torch包
  • 3. mmsegmention等安装

【注:从1开始进入正题】

小白关于环境配置的一点理解

linux下所有的命令行都是工具,类似软件,安装在某个位置,并把位置添加到环境变量里面。通过which XX就能查看到该位置在哪

nvidia-smi是一个管理监控GPU的工具

nvcc是运行时,编译驱动GPU的工具,安装cuda时一定会自带这个,所以检查这个就知道有没有CUDA

  • 两个不一样,后者输出的是runtime API,这个是运行时要调用的所以torch必须和这个配合。前者是驱动支持的最高版本的cuda,并不是实际安装的cuda

下载——(安装的位置)及配置其对应环境变量

  • 首先要知道在哪里安装:用find [路径] [匹配条件]去寻找。

  • 其次看环境变量里面有没有路径:

    • echo $PATH

    • 以及,环境变量都在当下shell的脚本里。比如~/.bashrc.通过vim XX就能看到

      • 如何添加环境变量,修改,这里有:Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_ubuntu20.04 cuda安装-CSDN博客

装环境要注意,因为依赖,有的顺序不能变。

0.装环境之前你需要知道有关CUDA的事

查看官网安装教程,我们首先发现,要区分以下几个:

OS:你使用GPU所用的操作系统

NVIDIA hardware:GPU物理意义上的显卡

NVIDIA CUDA:加速GPU显卡使用的(用c++写的)库

和他的驱动CUDA driver:使用CUDA需要的编译驱动.即nvcc

(下载CUDA包把这两个都包括了)

CUDNN:
- 用于神经网络的GPU加速库
- CUDA是工作台,有很多工具,cuDNN是其中一个。要在CUDA上运行深度神经网络,就要安装cuDNN,使GPU进行神经网络工作。

所以,要为深度学习配置环境,要安装CUDA及其驱动,cuDNN,pytorch(版本要对应才能成功)

进一步理解(optional)

打开官网给出的表格Support Matrix :: NVIDIA cuDNN Documentation,最右边的Maxwell, pascal等属于GPU架构。你可以

发布者:admin,转转请注明出处:http://www.yc00.com/web/1754955542a5220990.html

相关推荐

发表回复

评论列表(0条)

  • 暂无评论

联系我们

400-800-8888

在线咨询: QQ交谈

邮件:admin@example.com

工作时间:周一至周五,9:30-18:30,节假日休息

关注微信