文章目录
该文章收录于
配置Yolov5其实没那么复杂——Yolov5配置环境总结版
https://blog.csdn.net/hh571050143/article/details/119968613
打开cmd,激活相应环境
如下
python
import torch
a = torch.tensor(1.)
a.cuda()
from torch.backends import cudnn
cudnn.is_available()
cudnn.is_acceptable(a.cuda())