当前位置： 首页 > news >正文

几个常用线程的

news 来源：原创 2024/5/20 17:02:51
1
OMP_NUM_THREADS=1
export OMP_NUM_THREADS

在服务器上跑PyTorch 程序的列位不要把cpu占满
修改这个环境变量降低你们的cpu使用率
（因为pytorch默认是能用多少线程就用多少的）

在python代码中:

# 选择显卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# 只使用一个线程
os.environ["OMP_NUM_THREADS"] = "1"
2

这种方法经常失效，建议使用1中的方法
pytorch中设置线程个数
torch.set_num_threads(int)

3

选择用哪个显卡来跑程序
CUDA_VISIBLE_DEVICES=1 python xx.py

相关文章：

  • 可重入函数与不可重入函数
  • C#关于MSMQ通过HTTP远程发送专有队列消息的问题
  • python操作redis
  • Poisson Blending（Seamless clone)研究和实现
  • 【div】给div添加滚动条
  • DockOne微信分享( 九十四）：唯品会基于Kubernetes的网络方案演进
  • jenkins自动构建
  • Expression Tree上手指南 （一）转
  • 进入编辑模式 ， vim命令模式 ， vim实践
  • spring cloud 学习(10) - 利用springfox集成swagger
  • java.nio.channels.IllegalBlockingModeException
  • Eclipse在线更新慢
  • 【杂谈】小记一个ios11的bug
  • libgdx游戏引擎教程
  • android用户界面之WebView教程实例汇总
  • 【翻译】Mashape是如何管理15000个API和微服务的（三）
  • 【跃迁之路】【585天】程序员高效学习方法论探索系列（实验阶段342-2018.09.13）...
  • 0基础学习移动端适配
  • 11111111
  • HTTP中GET与POST的区别 99%的错误认识
  • Java IO学习笔记一
  • JavaScript HTML DOM
  • JS基础之数据类型、对象、原型、原型链、继承
  • js算法-归并排序（merge_sort）
  • Rancher-k8s加速安装文档
  • Shadow DOM 内部构造及如何构建独立组件
  • Vue 重置组件到初始状态
  • webpack4 一点通
  • 对JS继承的一点思考
  • 复杂数据处理
  • 将 Measurements 和 Units 应用到物理学
  • 线性表及其算法（java实现）
  • 再次简单明了总结flex布局，一看就懂...
  • #我与Java虚拟机的故事#连载09：面试大厂逃不过的JVM
  • (4) PIVOT 和 UPIVOT 的使用
  • （4.10~4.16）
  • （八）Docker网络跨主机通讯vxlan和vlan
  • （动态规划）5. 最长回文子串 java解决
  • （附源码）ssm高校升本考试管理系统 毕业设计 201631
  • (汇总)os模块以及shutil模块对文件的操作
  • （四）图像的%2线性拉伸
  • (一)kafka实战——kafka源码编译启动
  • （一）Thymeleaf用法——Thymeleaf简介
  • (一)为什么要选择C++
  • （原+转）Ubuntu16.04软件中心闪退及wifi消失
  • (转) 深度模型优化性能 调参
  • （转）PlayerPrefs在Windows下存到哪里去了？
  • .apk文件，IIS不支持下载解决
  • .NET/C# 中设置当发生某个特定异常时进入断点（不借助 Visual Studio 的纯代码实现）
  • .Net下使用 Geb.Video.FFMPEG 操作视频文件
  • /etc/apt/sources.list 和 /etc/apt/sources.list.d
  • ?php echo $logosrc[0];?,如何在一行中显示logo和标题？
  • [100天算法】-不同路径 III（day 73）
  • [2019.3.5]BZOJ1934 [Shoi2007]Vote 善意的投票
  • [Arduino学习] ESP8266读取DHT11数字温湿度传感器数据