展示 6 篇文章   每页 1
AI
1149 阅读

配置CUDA用于windows下yolo训练

换了一台windows,买了3060显卡,打算用这个卡试试训练,毕竟纯cpu训练太慢了。下面记录一下步骤。 先打开nvidia控制面板,点击右下角任务栏上能看到。再点击左下角的系统信息-组件,检查下支持的cuda版本。(要注意先检查pytorch官网里提示的能支持的cuda的最大版本,我......
python 深度学习 YOLO pytorch