查看GPU信息

最新推荐文章于 2024-10-16 11:02:34 发布

Mr_FengT

最新推荐文章于 2024-10-16 11:02:34 发布

阅读量3.2k

点赞数 5

分类专栏： pytorch

本文链接：https://blog.csdn.net/Mr_FengT/article/details/91564990

版权

pytorch 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

import torch
import os

#设置使用那块GPU “0”是GPU的id
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

#查看GPU是否可用
cuda_available=torch.cuda.is_available()  

print('cuda_available:',cuda_available) 

#查看可用GPU数量
coun=torch.cuda.device_count()
print(coun)

#返回GPU名字 0 是返回GPU名字的ID
name=torch.cuda.get_device_name(0)
print(name)
#返回当前设备的ID
num=torch.cuda.current_device()
print(num)

更多可以这儿https://pytorch.org/docs/stable/cuda.html