【问题】如何在docker中正确使用GPU?

报错

AssertionError: Torch not compiled with CUDA enabled

分析

错误表明 PyTorch 版本没有启用 CUDA 支持,因此无法使用 GPU 加速

解决方法