# 这是一个示例 Python 脚本。

# 按 Shift+F10 执行或将其替换为您的代码。
# 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。
import torch
import torchvision

print(torch.__version__)
print(torch.version.cuda)

print(torchvision.__version__)
print(torchvision.version.cuda)

# def print_hi(name):
#     # 在下面的代码行中使用断点来调试脚本。
#     print(f'Hi, {name}')  # 按 Ctrl+F8 切换断点。
#
#
# # 按间距中的绿色按钮以运行脚本。
# if __name__ == '__main__':
#     print_hi('PyCharm')

# 访问 https://www.jetbrains.com/help/pycharm/ 获取 PyCharm 帮助

print(torch.cuda.is_available())


# yolo task=detect mode=predict model=yolov8n.pt source=run-people.jpg imgsz=640 show=True save=True

# 训练
# yolo task=segment mode=train model=tof3D/yolov8-seg.yaml data=tof3D/data.yaml epochs=100 batch=4