Azure Ubuntu虚拟机无法识别NVIDIA GPU/CUDA问题求助

1. 根本问题：实例规格错误 #

D4s_v3是纯CPU实例，不带NVIDIA GPU，这是无法识别显卡的核心原因。必须更换为Azure N系列GPU实例（如NC6s_v3、ND40s_v3等，根据训练需求选择对应规格）。

2. 更换GPU实例后的正确操作 #

2.1 清理错误安装的组件 #

先卸载之前手动安装的不兼容工具包：

sudo apt purge nvidia-cuda-toolkit -y
sudo apt autoremove -y
sudo apt clean

2.2 安装官方GPU驱动 #

你使用的是NVIDIA GPU-Optimized镜像，Azure提供了一键安装脚本，直接执行：

sudo /opt/nvidia/nvaid_installer.sh

执行完成后重启虚拟机：

sudo reboot

2.3 验证GPU与CUDA状态 #

重启后运行以下命令确认：

• 查看GPU硬件信息：

nvidia-smi

• 检查CUDA版本：

nvcc --version

3. 适配TensorFlow环境 #

确保TensorFlow版本与当前CUDA版本兼容，例如CUDA 11.8可安装TensorFlow 2.13及以上版本：

pip install tensorflow==2.13.*

内容的提问来源于stack exchange，提问作者José Daniel Sarmiento