最开始，创建环境，用这个代码查看显卡 import torchprint(torch.cuda.is_available())print(torch.version.cuda)print(torch.cuda.get_devi