torch.set_num

t.set_num_threads 可以设置 PyTorch 进行 CPU 多线程并行计算时所占用的线程数，用来限制 PyTorch 所占用的 CPU 数目；
t.set_printoptions可以用来设置打印 tensor 时的数值精度和格式；

t.set_num_threads 在linux和Windows平台都有效：

linux有效，不用时cpu占用能到5000%，设置3后，就到到300%。

linux中，5000%代表用了50个cpu核，300%代表用了3个cpu核。

不设置，cpu51%，时间15ms

不设置的时候，torch会自动加载很多cpu，导致CPU占用很高。

2的时候，cpu17%,时间15ms变为25ms

4的时候，cpu34%，时间17ms

8的时候，cpu67%，

import timeimport torch
import numpy as npwhile True:torch.set_num_threads(4)arr= np.zeros((100,3, 300, 300), dtype=np.float32)# start=time.time()# arr=arr-0.406# print('time1',time.time()-start)aaa=torch.from_numpy(arr)start = time.time()aaa.add_(-0.406)print('time2',time.time() - start)# aaa.add_(-0.406)# print(aaa)# inps = torch.zeros(50, 3, 300, 300)time.sleep(0.007)

torch.set_num_threads相关推荐

PyTorch 笔记（12）— Tensor 持久化、向量化、torch.set_num_threads、torch.set_printoptions
1. 持久化在 PyTorch中 ,以下对象可以持久化到硬盘,并能通过相应的方法加载到内存中: Tensor Variable nn.Module Optimizer 本质上上述这些信息最终都是保存 ...
python torch exp_PyTorch官方中文文档：torch
torch 包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作.另外,它也提供了多种工具,其中一些可以更有效地对张量和任意类型进行序列化. 它有CUDA 的对应实现,可以在NVIDIA ...
PyTorch 重磅更新，不只是支持 Windows
翻译 | 林椿眄出品 | AI 科技大本营(公众号ID:rgznai100) 这次版本的主要更新一些性能的优化,包括权衡内存计算,提供 Windows 支持,24个基础分布,变量及数据类型,零维张量 ...
pytorch cpu占用较高
方法一.torch.set_num_threads(int thread) (亲测比较有效) linux有效,不用时cpu占用能到5000%,设置3后,就到到300%. 法二.export OMP_N ...
pytorch 归一化与反归一化
ToTensor中就有转到0-1之间了. # -*- coding:utf-8 -*-import timeimport torchfrom torchvision import transforms ...
pytorch 多进程队列
import os import time import torchimport torch.multiprocessing as mpfrom torch.multiprocessing impor ...
pytorch初入门学习记录
PyTorch第一步 1.Tensor from __future__ import print_function import torch as t # 构建 5x3 矩阵,只是分配了空间,未初始化 ...
python反归一化_pytorch 归一化与反归一化实例
ToTensor中就有转到0-1之间了. # -*- coding:utf-8 -*- import time import torch from torchvision import transfo ...
pytorch常见问题之cpu占满
问题: 在服务器上跑pytorch服务,cpu核心全部占用,影响其他服务的性能解决方法: pytorch默认是能用多少线程就用多少的,可通过一下方法设置线程修改环境变量 export OMP_NU ...

torch.set_num_threads

torch.set_num_threads相关推荐

最新文章

热门文章