博客
关于我
torch 查看GPU
阅读量:236 次
发布时间:2019-03-01

本文共 282 字,大约阅读时间需要 1 分钟。

import torch

查看cuda是否可用

print(torch.cuda.is_available())

查看cuda设备的数量

print(torch.cuda.device_count())

查看当前使用的cuda编号

print(torch.cuda.current_device())

查看GPU设备名字

print(torch.cuda.get_device_name())

查看设备容量

print(torch.cuda.get_device_capability(0))
return:True10GeForce GTX 1060(6, 1)

转载地址:http://kbbt.baihongyu.com/

你可能感兴趣的文章
MySQL单实例或多实例启动脚本
查看>>
MySQL压缩包方式安装,傻瓜式教学
查看>>
MySQL原理、设计与应用全面解析
查看>>
MySQL原理简介—1.SQL的执行流程
查看>>
MySQL原理简介—10.SQL语句和执行计划
查看>>
MySQL原理简介—11.优化案例介绍
查看>>
MySQL原理简介—12.MySQL主从同步
查看>>
MySQL原理简介—2.InnoDB架构原理和执行流程
查看>>
MySQL原理简介—3.生产环境的部署压测
查看>>
MySQL原理简介—6.简单的生产优化案例
查看>>
MySQL原理简介—7.redo日志的底层原理
查看>>
MySQL原理简介—8.MySQL并发事务处理
查看>>
MySQL原理简介—9.MySQL索引原理
查看>>
MySQL参数调优详解
查看>>
mysql参考触发条件_MySQL 5.0-触发器(参考)_mysql
查看>>
MySQL及navicat for mysql中文乱码
查看>>
MySqL双机热备份(二)--MysqL主-主复制实现
查看>>
MySQL各个版本区别及问题总结
查看>>
MySql各种查询
查看>>
mysql同主机下 复制一个数据库所有文件到另一个数据库
查看>>