配置CUDA开发环境

目录

相关概念

如何安装CUDA

方法一:安装GPU Driver

下载并安装GPU Driver

查看CUDA Driver Version

更新GPU Driver 

方法二:安装CUDA toolkit

下载并安装CUDA toolkit

查看CUDA Driver Version

查看CUDA编程环境

如何安装cuDNN

下载

安装

验证是否安装成功

查询CUDA toolkit与GPU Driver 的版本对应关系

查询GPU算力


相关概念

  • GPU:独立显卡,是一个硬件。需要安装显卡驱动(GPU Driver)才可使用;
  • CUDA:并行计算框架。用户可通过CUDA的API,调度GPU加速计算。包含CUDA驱动(CUDA Driver)和CUDA运行时库(CUDA Runtime)两部分。二者版本需满足如下关系:CUDA Runtime Version\leqCUDA Driver Version;
  • cudnn:神经网络训练和推理的加速库,用户可通过cudnn的API搭建神经网络并进行推理,再通过cuda调度GPU进行计算(也可以不使用cudnn,直接使用cuda搭建神经网络,但是运算效率会低很多);

如何安装CUDA

  • GPU Driver安装包:包含GPU Driver、CUDA Driver;
  • CUDA toolkit安装包:包含GPU Driver、CUDA Driver、CUDA Runtime、CUDA-C/C++编译器、一些科学库等CUDA开发工具。
  • 如果只是进行深度学习的模型训练、推理,那么安装GPU Driver安装包即可。如果需要在Visual Studio等平台上进行CUDA编程,则需要安装CUDA toolkit安装包。

方法一:安装GPU Driver

下载并安装GPU Driver

 官网下载NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA,输入电脑配置,搜索并下载驱动程序

按照默认设置安装完成后,重启电脑

查看CUDA Driver Version

按照如下方法,可查看CUDA Driver Version,说明安装成功。

方法1:win+r,输入cmd,打开命令窗口,输入:nvidia-smi。如下图所示:

Driver Version(GPU Driver Version):552.44

CUDA Version(CUDA Driver Version):12.4

方法2:打开NVIDIA控制面板—帮助—系统信息

如下图所示:GPU Driver Version:552.44

 CUDA Driver Version:12.4

更新GPU Driver 

卸载原始驱动程序(卸载后重建电脑,否则无法安装新驱动程序)

方法1:打开电脑的设置—应用,卸载原始驱动程序。

方法2:打开设备管理器,选中显卡,右键卸载设备。

           

下载并安装新驱动程序。

方法二:安装CUDA toolkit

下载并安装CUDA toolkit

CUDA Toolkit Archive | NVIDIA Developer

查看CUDA Driver Version

同上

查看CUDA编程环境

打开Visual Studio,新建项目。如下图所示,已经包含CUDA。

如何安装cuDNN

下载

 官网下载地址。需要注意的是:根据已经安装的CUDA Driver Version选择cuDNN安装包

安装

cuDNN安装包就是一个压缩包,解压后如下图所示:

将以上三个文件夹里的内容分别拷贝到CUDA安装目录对应的文件夹中 

验证是否安装成功

网上大部分人按照如下方式验证。本人亲测,不安装cuDNN,如下方式也能验证通过。所以如下方式不能验证cudnn是否安装成功,暂时我也不知道如何验证。

找到下图中的两个exe文件。

先打开cmd命令窗口,然后将这两个文件分别拖动到窗口内执行,弹出下列对话框则说明安装成功 。(如果在文件夹内,直接双击运行,则弹框会一闪而过)

查询CUDA toolkit与GPU Driver 的版本对应关系

CUDA 12.4 Update 1 Release Notes

查询GPU算力

 CUDA GPUs - Compute Capability | NVIDIA Developer

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
在Linux上配置CUDA开发环境需要一些步骤。以下是一般的配置步骤: 1. 检查您的显卡是否支持CUDA:首先,请确保您的显卡支持CUDA。您可以在NVIDIA官方网站上查找与您的显卡型号兼容的CUDA版本。 2. 安装显卡驱动:在Linux上使用CUDA需要安装适当的NVIDIA显卡驱动程序。您可以从NVIDIA官方网站下载适合您的显卡型号的驱动程序,并按照其文档进行安装。 3. 安装CUDA Toolkit:CUDA Toolkit包含了用于编译和运行CUDA代码所需的工具和库。您可以从NVIDIA官方网站下载适合您的操作系统和显卡型号的CUDA Toolkit,并按照其文档进行安装。 4. 设置环境变量:安装完成后,需要设置一些环境变量,以便系统能够正确找到CUDA工具和库。您可以将以下行添加到您的`.bashrc`或`.bash_profile`文件中: ```shell export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 5. 验证安装:输入命令 `nvcc --version` 检查CUDA是否成功安装并显示版本信息。 6. 编写和编译CUDA代码:使用任何文本编辑器编写CUDA代码,并使用nvcc编译器编译。例如,将代码保存为`example.cu`,然后运行以下命令进行编译: ```shell nvcc -o example example.cu ``` 7. 运行CUDA应用程序:编译成功后,您可以运行生成的可执行文件,例如: ```shell ./example ``` 以上是一般的步骤来配置CUDA开发环境。具体的步骤可能因您的Linux发行版和CUDA版本而有所不同。请参考NVIDIA官方文档以获得更详细的指南。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

进击的路飞桑

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值