以下是Ubuntu下PyTorch的調試技巧:
torch.cuda.is_available()
驗證CUDA是否可用。import pdb; pdb.set_trace()
設置斷點,支持單步執行、查看變量等操作。logging
模塊記錄程序狀態。torch.autograd.profiler
分析計算性能。torch.autograd.set_detect_anomaly(True)
檢測梯度異常。unittest
或pytest
編寫單元測試。pylint
或flake8
檢查代碼質量。