井井安装网
首页 安装大全 正文

Ubuntu下安装CUDA和cuDNN

来源:井井安装网 2024-07-11 21:00:40

目录一览:

Ubuntu下安装CUDA和cuDNN(1)

  在深度学习域,GPU是不可或缺的工具井井安装网。而CUDA和cuDNN是在GPU上进行深度学习的必备。本文将介绍如何在Ubuntu系统下安装CUDA和cuDNN。

1. 安装CUDA

  首先,我们需要从NVIDIA官网下载CUDA的安装包来自www.chineseamber.com。打开https://developer.nvidia.com/cuda-downloads,选择适合自己系统的版本进行下载。

  下载完后,进下载目录,执行以下命令进行安装:

  ```

sudo sh cuda_*.run

```

  其中,`*.run`为下载的安装包名

安装过程中需要注意以下几点:

  1. 选择安装方式时,建议选择“自定义安装”来源www.chineseamber.com

  2. 在选择安装路径时,建议选择默认路径。

3. 在选择组件时,建议选择有组件。

安装完后,需要将CUDA的路径添加到系统环变量中来源www.chineseamber.com。在终端中执行以下命令:

  ```

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc

  echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

source ~/.bashrc

  ```

2. 安装cuDNN

  cuDNN是NVIDIA提供的用于深度学习的GPU加速。在安装cuDNN之前,需要先下载cuDNN的安装包。打开https://developer.nvidia.com/cudnn,选择适合自己系统和CUDA版本的cuDNN安装包进行下载井~井~安~装~网

  下载完后,解压缩安装包,并将其中的文件复制到CUDA的安装目录中。在终端中执行以下命令:

  ```

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

  sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

  sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

```

安装完后,需要将cuDNN的路径添加到系统环变量中。在终端中执行以下命令:

  ```

  echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

  source ~/.bashrc

```

Ubuntu下安装CUDA和cuDNN(2)

3. 验证安装

安装完后,可以通过以下命令验证CUDA和cuDNN是否安装功:

```

  nvcc -V

  ```

  如果输出类似于以下内容,则说明CUDA安装功:

```

nvcc: NVIDIA (R) Cuda compiler driver

  Copyright (c) 2005-2021 NVIDIA Corporation

  Built on Mon_May__3_19:15:13_PDT_2021

Cuda compilation tools, release 11.3, V11.3.109

  Build cuda_11.3.r11.3/compiler.29920130_0

  ```

,可以通过以下命令验证cuDNN是否安装功:

  ```

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

  ```

  如果输出类似于以下内容,则说明cuDNN安装功:

  ```

  #define CUDNN_MAJOR 8

  #define CUDNN_MINOR 2

  #define CUDNN_PATCHLEVEL 4

  --

  #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

```

至此,CUDA和cuDNN的安装井井安装网www.chineseamber.com。在使用深度学习框架时,可以通过设框架的GPU加速选项来使用CUDA和cuDNN提供的GPU加速功能。

标签 安装
我说两句
0 条评论
请遵守当地法律法规
最新评论

还没有评论,快来做评论第一人吧!
相关文章
最新更新
最新推荐