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

火山方舟向量数据库大模型

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

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

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

nvidia-smi

picture.image

这个命令会显示电脑的 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)和你的操作系统。

picture.image

下载对应的安装包。

picture.image

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

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

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

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

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

复制代码

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp

验证安装:

picture.image

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

nvcc --version

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

picture.image

最后,要安装 cuDNN

访问 NVIDIA cuDNN 下载页面。

https://developer.nvidia.com/cudnn

picture.image

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

picture.image

然后安装 cuDNN 。

0
0
0
0
关于作者

文章

0

获赞

0

收藏

0

相关资源
字节跳动客户端性能优化最佳实践
在用户日益增长、需求不断迭代的背景下,如何保证 APP 发布的稳定性和用户良好的使用体验?本次分享将结合字节跳动内部应用的实践案例,介绍应用性能优化的更多方向,以及 APM 团队对应用性能监控建设的探索和思考。
相关产品
评论
未登录
看完啦,登录分享一下感受吧~
暂无评论