怎么用python代码查看可用的gpu,然后指定可用的gpu运行

查看gpu

python 复制代码
import torch

# 检查CUDA是否可用
if torch.cuda.is_available():
    # 获取GPU设备数量
    device_count = torch.cuda.device_count()

    # 列出可用的GPU设备
    for i in range(device_count):
        print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
else:
    print("CUDA is not available. No GPU devices found.")

输出结果

设置指定gpu运行

指定上面输出的5号卡

python 复制代码
if torch.cuda.is_available():
    # 指定要使用的GPU设备编号
    device = torch.device("cuda:5")
    print(f"Using GPU {device} - {torch.cuda.get_device_name(device)}")
else:
    print("CUDA is not available. No GPU devices found.")

查看gpu的内存情况等

python 复制代码
import subprocess

# 执行nvidia-smi命令以获取GPU信息
nvidia_smi_output = subprocess.check_output("nvidia-smi", shell=True).decode()

# 切分输出为每个GPU的信息
gpu_info = nvidia_smi_output.strip().split('\n\n')

# 遍历每个GPU的信息
for i, info in enumerate(gpu_info):
    print(f"GPU {i}:")
    print(info)
相关推荐
来一份培根4 分钟前
Win10安装GCC/G++运行环境
python·pointnet工具
新加坡内哥谈技术19 分钟前
Llama 4 家族:原生多模态 AI 创新新时代的开启
人工智能·深度学习·语言模型·自然语言处理·自动化·llama
暴龙胡乱写博客22 分钟前
python三大库之---pandas(二)
开发语言·人工智能·python·pandas
winfredzhang41 分钟前
博客文章:深入分析 PyMovie - 基于 Python和 MoviePy 的视频管理工具
python·ffmpeg·音视频·pymovie
独好紫罗兰1 小时前
洛谷题单3-P1307 [NOIP 2011 普及组] 数字反转-python-流程图重构
开发语言·python·算法
墨绿色的摆渡人1 小时前
具身智能零碎知识点(一):深入解析Transformer位置编码
人工智能·深度学习·transformer
牙牙要健康1 小时前
【深度学习】【目标检测】【Ultralytics-YOLO系列】YOLOV3源码整体结构解析
深度学习·yolo·目标检测
蹦蹦跳跳真可爱5891 小时前
Python----计算机视觉处理(Opencv:道路检测完整版:透视变换,提取车道线,车道线拟合,车道线显示,)
开发语言·人工智能·python·opencv·计算机视觉
帮帮志2 小时前
Python代码list列表的使用和常用方法及增删改查
开发语言·python
小洛~·~3 小时前
《深度学习入门:基于Python的理论与实现》第三章神经网络
python·深度学习·神经网络