Holen Sie sich die Version von CUDA in Pytorch

torch.version.cuda
Pouyan