安装完pytorch
import torch print(torch.__version__) print(torch.cuda.is_available())
特意调用代码看了下安装成功了，当时返回的是‘true’ 显示cuda安装成功了的可是问题一下子就出现了，当我安装完 pycharm 后运行yolo5 在用上面代码是 突然发现 显示竟然变成了‘false’
因此训练一直用的cpu再跑，我已经重装几次，还是这个问题
那你的pycharm 用的环境和你安装的环境是一个环境吗？