官方 docker ubuntu20.04 里的工具链 gpu 版本是否支持 5060显卡

开发与问题
1

我感觉 cuda 版本太低，5060 好像用不了

2

是否有支持 NVIDIA Blackwell、RTX 50系列、CUDA 12.x及计算能力sm_120的RDK X5 OpenExplorer GPU工具链镜像？当前公开v1.2.8镜像为CUDA 11.6、PyTorch 1.13，仅编译到sm_86，无法在RTX 5060上执行CUDA算子。

3

目前提供的工具链镜像暂时没有更新计划，一般来说只有进行量化感知训练（QAT）时才会调用GPU资源。因此如果您只是对模型进行训练后量化（PTQ）操作的话，使用CPU版本工具链即可。