企业官网建设流程全解析

linux下的CUDA安装

最终效果需要能够实现三个命令

nvidia-smi

看到显卡

nvcc--version

看到CUDA版本，通常是12.1

python -<<'PY' import torch print(torch.cuda.is_available()) print(torch.cuda.get_device_name(0)) PY

输出为

True 你的 NVIDIA 显卡型号

1查显卡型号

在window下
方法一
打开任务管理器
性能 → GPU

方法二
PowerShell 里面执行

nvidia-smi

如果 NVIDIA 驱动正常，会显示类似：

+---------------------------------------------------------------------------------------+|NVIDIA-SMI552.22Driver Version:552.22CUDA Version:12.4||GPU Name...|0NVIDIA GeForce RTX5060... +---------------------------------------------------------------------------------------+

你要记三个东西：

GPU Name Driver Version CUDA Version

2在ubuntu安装

2.1先确定ubuntu能识别到显卡

lspci|grep-invidia

如果有.

01:00.0 VGA compatible controller: NVIDIA Corporation GA106[GeForce RTX3060]

说明 Ubuntu 识别到硬件。

如果没输出，可能是：

• BIOS 没开独显；
• 笔记本混合显卡设置问题；
• 驱动还没装但一般 lspci 仍应能看到；
  2.2 安装驱动
  方案A：自动安装驱动
  先能拉下载的驱动包

sudoaptupdatesudoaptupgrade-ysudoaptinstall-yubuntu-drivers-common

可以查看推荐的驱动

ubuntu-drivers devices

然后自动安装

sudoubuntu-drivers autoinstall

安装完重启

sudoreboot

重启后检查

nvidia-smi

正常应该显示：

NVIDIA-SMI xxx.xx Driver Version: xxx.xx CUDA Version:12.x GPU Name: NVIDIA GeForce...

如果nvidia-smi报错了！！！

NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

大概率是驱动没加载，常见原因是 Secure Boot。
重启电脑，进 BIOS，把：

Secure Boot

设置为

Disabled

方案B：根据型号安装
1.鄙人的在window里面查的信息为

NVIDIA GeForce RTX5060Laptop GPU Windows 驱动：582.05 Windows 显示 CUDA Version：13.0

需要在双系统中安装

NVIDIA Driver：580+ CUDA Toolkit：12.8

2.进入ubuntu先查系统和显卡

lspci|grep-invidia

根据型号找到需要对应安装的驱动
3.安装NVIDIA驱动
更新系统

sudoaptupdatesudoaptupgrade-y

添加 NVIDIA 驱动 PPA
因为 RTX 5060 很新，建议加 PPA 获取更新驱动：

sudoadd-apt-repository-yppa:graphics-drivers/ppasudoaptupdate

查看系统推荐驱动

ubuntu-drivers devices

你要优先装带 recommended 的那个
根据推荐结果安装驱动
情况 A：看到 nvidia-driver-580 推荐

sudoaptinstall-ynvidia-driver-580

情况 B：看到 nvidia-driver-580-open 推荐

sudoaptinstall-ynvidia-driver-580-open

重启并检查驱动

在这里插入代码片

sudoreboot

重启进入 Ubuntu 后执行：

nvidia-smi

笔记本双显卡设置

系统里可能还有 Intel 核显
如果 nvidia-smi 正常，但 Python 里找不到 GPU，可以安装并设置 prime：

sudoaptinstall-ynvidia-prime

切到 NVIDIA 模式：

sudoprime-select nvidiasudoreboot

安装 CUDA Toolkit

根据Ubuntu二选一，注意根据自己的CUDA来选择cuda-toolkit-12-8
Ubuntu 22.04 安装 CUDA 12.8

cd~wgethttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.debsudodpkg-icuda-keyring_1.1-1_all.debsudoaptupdatesudoaptinstall-ycuda-toolkit-12-8

Ubuntu 24.04 安装 CUDA 12.8

cd~wgethttps://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.debsudodpkg-icuda-keyring_1.1-1_all.debsudoaptupdatesudoaptinstall-ycuda-toolkit-12-8

配置 CUDA 环境变量

echo'export CUDA_HOME=/usr/local/cuda-12.8'>>~/.bashrcecho'export PATH=/usr/local/cuda-12.8/bin:$PATH'>>~/.bashrcecho'export LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64:$LD_LIBRARY_PATH'>>~/.bashrcsource~/.bashrc

检查

nvcc--version

正常应该看到

Cuda compilation tools, release12.8