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