print(torch.version.cuda) print(torch.backends.cudnn.version())”, the output were: 0.4.1.post2 (PyTorch version) 9.0.176 (CUDA version)
確定! 回上一頁