博客
关于我
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从入门到精通(二)
查看>>
mysql以服务方式运行
查看>>
mysql优化--索引原理
查看>>
MySQL优化配置详解
查看>>
mysql会员求积分_MySql-统计所有会员的最高前10次的积分和
查看>>
MySQL修改密码报错ERROR 1396 (HY000): Operation ALTER USER failed for ‘root‘@‘localhost‘
查看>>
Mysql全局优化参数
查看>>
MySQL全面瓦解:安装部署与准备
查看>>
MySQL内存表使用技巧
查看>>
MySQL函数
查看>>
mysql函数汇总之数学函数
查看>>
mysql函数汇总之条件判断函数
查看>>
mysql函数汇总之系统信息函数
查看>>
MySQL函数简介
查看>>
mysql函数遍历json数组
查看>>
MySQL函数(转发)
查看>>
mysql分区表
查看>>