支持情况与前提条件
快速验证是否可用
nvidia-smi,应显示GPU型号、驱动版本、CUDA版本等信息。import torch; print(torch.cuda.is_available())print(torch.cuda.device_count())print(torch.cuda.get_device_name(0))torch.cuda.is_available()返回True,说明PyTorch已成功启用GPU加速。安装要点
sudo ubuntu-drivers autoinstall && sudo rebootsudo apt install nvidia-driver-<版本号>。pip install torch ... --extra-index-url https://download.pytorch.org/whl/cu118 或 conda install pytorch cudatoolkit=11.8 -c pytorch)。pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorchtorch.cuda.is_available()。常见故障排查
nvidia-smi不可用或报错:重装/更新驱动,使用ubuntu-drivers autoinstall后重启。torch.cuda.is_available()返回False:
PATH、LD_LIBRARY_PATH)是否包含CUDA路径。