search

DataXujing / YOLO-v5

:art: Pytorch YOLO v5 训练自己的数据集超详细教程！！！ :art: (提供PDF训练教程下载）
GNU General Public License v3.0
913 stars 196 forks source link

无法使用GPU训练 #21

Open w682511 opened 4 years ago

w682511 commented 4 years ago

您好，我似乎无法使用GPU进行训练，我做了一些尝试但都没有奏效，很抱歉麻烦您，这或许是一个简单的问题，可他确实困扰了很久，希望可以得到您的帮助，告诉我我应该做什么更改呢？感谢您！

DataXujing commented 4 years ago

请检查你的Pytorch版本和CUDA,及显卡驱动的版本是否一致，并在python解释器或终端中执行：

import torch
torch.cuda.is_available()

如果是True, 则说明你的torch库和显卡的相关版本一致，torch可正常调用GPU，否则你安装的pytorch是有问题的！

附：

# torch版本
import torch
torch.__version__

# cuda版本
$ nvcc

# 显卡驱动版本
$ nvidia-smi -lms 200