在服务器上训练的网络放到本地台式机进行infer，结果出现报错：
AssertionError: Invalid device id
仔细检查后发现原来服务器有多个GPU，当时开启了两个进行加速运算。
net1 = nn.DataParallel(net1, device_ids=[0, 1])
而本地台式机只有一个GPU，调用数量超出所以报错。
改为
net1 = nn.DataParallel(net1, device_ids=[0])
问题解决。
以上这篇解决pytorch报错:AssertionError: Invalid device id的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
前言 图像颜色的反转,比较简单的思路就是使用255减去当前值,从而得到反转后的图像.原始图片: 1.灰度图像的颜色反转 import cv2 import numpy as n...
本文实例讲述了Python基于ThreadingTCPServer创建多线程代理的方法。分享给大家供大家参考，具体如下： #coding=utf8 from BaseHTTPServ...
使用HTMLTestRunner输出的测试报告中，标题和错误说明的中文乱码。 环境： python v3.6 HTMLTestRunner v0.8.2 定位问题 刚开始以为是pytho...
使用python + shell 编写，是一个简易solaris系统巡检程序 复制代码 代码如下:#!/usr/bin/python -u#-*- coding:utf-8 -*-''...
定制pattern的string模板(template) 详解 string.Template的pattern是一个正则表达式, 可以通过覆盖pattern属性, 定义新的正则表达式....