yolo v5,请问如图所示,我的电脑明明是有cuda,为什么在pycharm里却显示不可用呢?那我应该如何使用cuda去训练模型呢?
1条回答 默认 最新
- PellyKoo 2022-11-22 20:42关注
第一张图只能证明你安装了nvidia驱动,但可能未安装cuda,
如果已经安装cuda,就要安装torch的cuda版本,而不是cpu版本。
安装cuda版的torch命令应该是pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报