pytorch查看CUDA支持情况，只需要三行代码