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

[pytorch]torch.cuda用法以及判断显卡是不是存在问题

常见用法:

torch.cuda.is_available() # 查看是否有可用GPU
torch.cuda.device_count() # 查看GPU数量
torch.cuda.get_device_capability(device) # 查看指定GPU容量
torch.cuda.get_device_name(device) # 查看指定GPU名称
torch.cuda.empty_cache() # 清空程序占用的GPU资源
torch.cuda.manual_seed(seed) # 设置随机种子
torch.cuda.manual_seed_all(seed) # 设置随机种子
torch.cuda.get_device_properties(i) # i为第几张卡,显示该卡的详细信息
 

场景问题:我使用torch.cuda.device_count()返回1但是我用nvidia-smi显示是2个显卡,这个是为啥呢?

第一个原因:你在环境变量设置了CUDA_VISIBLE_DEVICES

第二个原因:你显卡坏了一个,如何判断是不是坏了可以使用上面接口测试

import torch
device=torch.device("cuda:0")
print(torch.cuda.get_device_capability(device))

把0改成1如果报错则表示1这个显卡有问题或者不存在,据此可以判断显卡坏了。但是这个只是系统层面表示坏了。还需要进一步判断。首先重启系统在测试一次,不行就把显卡拔插一下,依然不行则做最后尝试把系统重装一次(这个一般都没效果),还不行只能说明显卡坏了。

http://www.lryc.cn/news/142369.html

相关文章:

  • JUC——多线程补充
  • 代码随想录第32天|122.买卖股票的最佳时机 II,55. 跳跃游戏 ,45. 跳跃游戏 II
  • Linux:Nginx服务与搭建
  • 4、什么是NoSQL
  • 如何自己实现一个丝滑的流程图绘制工具(一)vue如何使用
  • ReoGrid.NET集成到winfrom
  • Elasticsearch实现增删改查
  • Rust 学习笔记(卷二)
  • android amazon 支付接入
  • Vue2-快速搭建pc端后台管理系统
  • 【产品文档】团队介绍PPT模板
  • 组件库的使用和自定义组件
  • 网站和API支持HTTPS,最好在Nginx上配置
  • UnitTest笔记: 拓展库DDT的使用
  • 裂缝检测,只依赖OPENCV,基于YOLO8S
  • python编程环境使用技巧3-程序打包pyinstaller
  • Go 自学:defer关键字
  • 【云计算】Docker特别版——前端一篇学会
  • .net使用RabbitMQ小记
  • layUI 中 穿梭框无法获取值的细节问题
  • Kaggle回归问题Mercedes——Benz Greener Manufacturing
  • 天润融通「微藤大语言模型平台2.0」以知识驱动企业高速增长
  • 【BUG】解决安装oracle11g或12C中无法访问临时位置的问题
  • 2. 使用IDEA创建Spring Boot Hello项目并管理依赖——Maven入门指南
  • Python在电路课程中的应用
  • Spark SQL join的三种实现方式
  • wazuh环境配置和漏洞复现
  • 九五从零开始的运维之路(其三十六)
  • 同步和异步有什么区别,使用场景?
  • webassembly009 transformers.js 网页端侧推理