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