基于Pytorch查看本地或者远程服务器GPU及使用方法
一、查看GPU具体信息
nvidia-smi # 查看GPU型号等具体信息
二、基于Pytorch查看本地或者远程服务器GPU
import torch
print(torch.cuda.is_available() )# cuda是否可用
print(torch.cuda.device_count() )# gpu数量
print(torch.cuda.current_device())# 当前设备索引, 从0开始
print(torch.cuda.get_device_name(0))# 返回gpu名字
三、GPU使用方法(典型)
device = torch.cuda.current_device() if args.cuda else torch.device('cpu')
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
os.environ['CUDA_VISIBLE_DEVICES']='0'
评论(1)
您还未登录,请登录后发表或查看评论