视觉组第七次培训

主要内容

人工智能 8.13

1. 神经网络
2. yolo 由cyc来讲

环境配置

安装显卡驱动（N卡）

https://www.nvidia.cn/geforce/drivers/
Windows大部分自带显卡驱动
Ubuntu需要自己下载

验证是否安装成功：

nvidia-smi

安装cuda

https://blog.csdn.net/YYDS_WV/article/details/137825313 (Windows)
https://blog.csdn.net/zxdd2018/article/details/127705627 (Ubuntu)

验证是否安装成功：

nvcc -V

安装pytorch-gpu

https://pytorch.org/get-started/locally/

验证是否安装成功：

import torch

# 检查CUDA是否可用
if torch.cuda.is_available():
    print(f"CUDA is available! Number of GPUs: {torch.cuda.device_count()}")
    print(f"Current GPU: {torch.cuda.get_device_name(torch.cuda.current_device())}")
else:
    print("CUDA is not available.")

安装ultralytics

注意要先安装pytorch-gpu再装yolo，否则只有cpu加速，可以运行上面的代码检测是否能检测到cuda

https://docs.ultralytics.com/quickstart/#install-ultralytics

标数据集网站

https://app.roboflow.com/