| 123456789101112131415161718192021222324252627282930313233 |
- # 这是一个示例 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
|
