程序找不到 GPU。可能是驱动没装、CUDA 没装、Docker 没挂 GPU。
nvidia-smi
docker run --gpus all ...
pip install torch --index-url https://download.pytorch.org/whl/cu121
torch.cuda.is_available()
本站為技術速查,內容僅供參考。錯誤訊息以官方文件為準。