【pytorch】 Major minor of gpu

个人笔记

msg = torch.cuda.get_device_properties(torch.cuda.current_device())
data = msg.__getattribute__("major")
data = msg.__getattribute__("minor")

Gpu支持的最高CUDA版本号和最低版本号

你可能感兴趣的:(人工智能(未)入门,pytorch,深度学习,python)