玩AI,如何查看电脑的 CUDA 和 GPU 信息并进行配置?

2024-07-31 16:29:23 浏览数 (1)

玩AI模型,GPU算力是基础,首先得了解自己电脑的 CUDA 和 GPU 信息,并安装相对应的软件、进行正确的配置。

要查看电脑的 CUDA 和 GPU 信息,可以通过命令行查看

查看 GPU 信息,打开命令提示符或 PowerShell,然后输入以下命令:

nvidia-smi

这个命令会显示电脑的 GPU 型号、驱动程序版本以及 CUDA 版本。根据 nvidia-smi 输出的信息, GPU 是 NVIDIA GeForce,并且当前安装的 CUDA 版本是 11.4。

查看 CUDA Toolkit 版本,打开命令提示符或 PowerShell,然后输入以下命令:

nvcc --version

这个命令会显示安装的 CUDA Toolkit 的版本信息。

如果没有安装 nvcc,你可能需要先安装 CUDA Toolkit。要在你的系统上安装 CUDA Toolkit,可以按照以下步骤进行:

下载 CUDA Toolkit 安装包:

https://developer.nvidia.com/cuda-toolkit-archive 访问 NVIDIA CUDA Toolkit 下载页面。

选择合适的版本(例如 CUDA 11.4)和你的操作系统。

下载对应的安装包。

安装 CUDA Toolkit:运行下载的安装包并按照提示完成安装。

在安装完成后,需要将 CUDA 的 bin 目录和 libnvvp 目录添加到系统的环境变量中。

打开“系统属性” -> “高级系统设置” -> “环境变量”。

在“系统变量”部分,找到 Path,点击“编辑”。

添加以下路径(根据你的安装路径进行调整):

复制代码

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.4bin

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.4libnvvp

验证安装:

打开命令提示符,输入以下命令来验证 CUDA Toolkit 是否安装成功:

nvcc --version

你应该能看到 CUDA 编译器的版本信息。

最后,要安装 cuDNN

访问 NVIDIA cuDNN 下载页面。

https://developer.nvidia.com/cudnn

登录并下载与 CUDA 版本匹配的 cuDNN 文件。

然后安装 cuDNN 。

0 人点赞