File size: 237 Bytes
9171a3a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# -*- coding: utf-8 -*-
"""

Author: Kungia

@Software: Pycharm

@file: train_test.py

@Author: 28963

@dateTime: 2025/8/5 17:22

@brief: 

"""


# import torch
# # print(torch.cuda.is_available())
# print(torch.version.cuda)