在服务器上训练的网络放到本地台式机进行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设计】。
如果对象是二维数组，则切片应当是x[:]的形式，里面有一个冒号，冒号之前和之后分别表示对象的第0个维度和第1个维度； 如果对象是三维数组，则切片应当是x[::]，里面有两个冒号，分割出...
MNIST数据集比较小，一般入门机器学习都会采用这个数据集来训练 下载地址：yann.lecun.com/exdb/mnist/ 有4个有用的文件： train-images-idx3...
工作中遇到了要计算两个数百分比的问题，python 2.7 环境。 代码： #!/usr/bin/env python #function: 计算百分比 #USAGE: python...
一、zipfile模块的简述 zipfile是python里用来做zip格式编码的压缩和解压缩的，由于是很常见的zip格式，所以这个模块使用频率也是比较高的， 在这里对zipfile的使...
Django 中自带了 sitemap框架，用来生成 xml 文件 Sitemap(站点地图)是通知搜索引擎页面的地址，页面的重要性，帮助站点得到比较好的收录。 白话文就是：一个写了你...