首先编译腾讯开源的ncnn:https://github.com/tongxiaobin/ncnn在tools/pytorch里最下
只有使用一块gpu训练的模型才可以使用该方法: 就是加载模型的时候在后面加上一个参数map_location='cpu'。然后把所有的.cuda()方法删掉。模型.load_state_dict(torch.load('训练好的模型路径',map_location='cpu'))
import torchtorch.cuda.is_available()#cuda是否可用;torch.cuda.device_count()#返回gpu数量;torch.cuda.get_device_name(0)#返回gpu名字,设备索引默认从0开始;torch.cuda.current_device()#返回当前设备索引;#另外在linux命令行中输入nvidia-smi可以查看GPU显
pycharm配置运行环境
pycharm新建工程
Copyright © 2005-2024 51CTO.COM 版权所有 京ICP证060544号