pytorch-gpu版本安装(英伟达gpu驱动安装)

一、安装cuda

1️⃣ 检查是否有 GPU

lspci | grep -i nvidia

如果没有输出，可能你的服务器 没有 GPU，或者 GPU 未正确识别。

2️⃣ 检查 NVIDIA 驱动是否安装

dpkg -l | grep -i nvidia

如果没有相关输出，说明驱动未安装，建议安装最新的 NVIDIA 驱动。

3️⃣ 重新安装 NVIDIA 驱动

如果驱动未正确安装，可以尝试重新安装：

sudo apt update
sudo apt install -y nvidia-driver-535  # 535 是示例版本，可换成最新版本

安装完成后，重启系统：

sudo reboot

4️⃣ 确保 NVIDIA 驱动已加载

执行：

lsmod | grep nvidia

如果没有输出，说明驱动未加载，可以手动加载：

sudo modprobe nvidia

5️⃣ 确保 NVIDIA 驱动服务已启动

sudo systemctl restart nvidia-persistenced

然后再尝试运行：

nvidia-smi

6️⃣ 检查 CUDA 是否安装

如果你要使用 CUDA，也需要检查：

nvcc --version

如果 CUDA 没有安装，可以从 NVIDIA 官网下载安装：

sudo apt install -y cuda

---

二、安装Pytorch

1️⃣查看cuda版本

2️⃣pytorch官网查找对应版本的安装命令行

pip3 install torch torchvision torchaudio

安装过程有点漫长😮‍💨😮‍💨😮‍💨