AssertionError: Invalid CUDA '--device gpu' requested, use '--device cpu' or pass valid CUDA device(s)

这错误通常是因为您的CUDA设备配置有误导致的。可能有以下几种情况：

1. 您的CUDA版本和您的PyTorch版本不兼容，建议您检查CUDA和PyTorch的版本是否匹配。
2. 您的CUDA驱动程序可能需要更新，建议您更新CUDA驱动程序。
3. 您的CUDA环境变量可能设置错误，建议您检查CUDA环境变量是否设置正确。
4. 您的CUDA设备可能不存在或者不可用，建议您检查您的CUDA设备是否存在并且可用。

以上是一些可能导致这个错误的原因，您可以针对这些原因逐一排查。