判断pytorch是否可用？输出GPU数量和编号

最新推荐文章于 2024-07-26 17:23:36 发布

在路上66

最新推荐文章于 2024-07-26 17:23:36 发布

阅读量3.7k

点赞数 4

分类专栏： python pytorch 文章标签： python cuda 深度学习 pytorch gpu

本文链接：https://blog.csdn.net/qq_18833967/article/details/117036434

版权

python 同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

pytorch

1 篇文章 0 订阅

订阅专栏

1.输出torch版本号

2.判断安装后的torch是否可用？

3.输出GPU数

4.输出GPU名字

5.输出现在正在使用的GPU编号

import torch


print("torch版本号:",end="")
print(torch.__version__)
print("判断torch是否可用:",end="")
print(torch.cuda.is_available())

print("gpu数量:",end="")
print(torch.cuda.device_count())

print("gpu名字，设备索引默认从0开始:",end="")
print(torch.cuda.get_device_name(0))
print("现在正在使用的GPU编号:",end="")
print(torch.cuda.current_device())