玩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 。