获取PC 服务器可用的GPU

雨点打透心脏的1/2处 2022-12-24 15:56 120阅读 0赞

### 获取PC 服务器 可用的GPU ###

*  代码
 *  结果展示

# 代码 #

def get_available_gpus():
        from tensorflow.python.client import device_lib as _device_lib
        local_device_protos = _device_lib.list_local_devices()
        return [x.name for x in local_device_protos if x.device_type == 'GPU']
    print(get_available_gpus())

# 结果展示 #

2020-11-29 15:45:53.923121: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
    2020-11-29 15:45:54.190561: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties: 
    name: GeForce GTX 1650 major: 7 minor: 5 memoryClockRate(GHz): 1.56
    pciBusID: 0000:01:00.0
    totalMemory: 4.00GiB freeMemory: 3.25GiB
    2020-11-29 15:45:54.190759: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1511] Adding visible gpu devices: 0
    ['/device:GPU:0']
    2020-11-29 15:45:55.004208: I tensorflow/core/common_runtime/gpu/gpu_device.cc:982] Device interconnect StreamExecutor with strength 1 edge matrix:
    2020-11-29 15:45:55.004314: I tensorflow/core/common_runtime/gpu/gpu_device.cc:988]      0 
    2020-11-29 15:45:55.004368: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1001] 0:   N 
    2020-11-29 15:45:55.004544: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/device:GPU:0 with 2941 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1650, pci bus id: 0000:01:00.0, compute capability: 7.5)
    
    Process finished with exit code 0

发表评论取消回复

表情：

评论列表（有 0 条评论，120人围观）

还没有评论，来说两句吧...

相关阅读

相关服务器使用指定的 GPU 训练神经网络(笔记)

简单记个笔记当我们用实验室服务器的多个 GPU 训练神经网络并指定特定的 GPU 时可以使用以下命令: 1、torch 1.9.0 之前的版本(不包括 1.9.0)

r囧r小猫/ 2024年04月01日 09:10/ 0 赞/ 47 阅读

相关部署GPU服务器的操作

部署GPU服务器的操作查看GPU型号 1. ==快捷键== 2. Notebook中

旧城等待，/ 2023年01月10日 13:26/ 0 赞/ 93 阅读

相关查询计算机上是否有可用的GPU设备

查询计算机上是否有可用的GPU设备代码结果展示代码 def is_gpu_available(cuda_only=True):

Myth丶恋晨/ 2022年12月24日 15:57/ 0 赞/ 92 阅读

相关获取PC 服务器可用的GPU

获取PC 服务器可用的GPU 代码结果展示代码 def get_available_gpus(): from tenso

雨点打透心脏的1/2处/ 2022年12月24日 15:56/ 0 赞/ 121 阅读

相关【tensorflow-2.x-gpu 】获得tf版本以及可用的设备CPU/GPU

【tensorflow-2.x-gpu 】获得tf版本以及可用的设备CPU/GPU 1.背景 2.代码示例 1.背景 tensorflow更新很快，

Dear 丶/ 2022年12月19日 14:29/ 0 赞/ 133 阅读

相关服务器常见高可用方案

前言后台服务可以划分为`两类`，`有状态`和`无状态`。高可用对于无状态的应用来说是比较简单的，无状态的应用，只需要通过 F5 或者任何代理的方式就可以很好的解决。后文

清疚/ 2022年11月09日 12:50/ 0 赞/ 417 阅读

相关本地PC连接远程服务器上的Tensorboard

转自： [ 本地PC连接远程服务器上的Tensorboard - 知乎][_PC_Tensorboard -] tensorboard --logdir

阳光穿透心脏的1/2处/ 2022年09月17日 03:14/ 0 赞/ 184 阅读

相关查看linux服务器gpu使用情况

> 指令：nvidia-smi ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmN

秒速五厘米/ 2022年03月20日 13:19/ 0 赞/ 324 阅读

相关配置远程GPU服务器

实验室刚好有一台GPU服务器，假期想远程使用，就想能否试一下像Google的CoLab那样通过jupyter notebook进行访问，试了一下确实可以，这里记录一下过程。服务

比眉伴天荒/ 2022年03月20日 09:25/ 0 赞/ 529 阅读

相关可用的时间服务器

202.120.2.101 133.100.9.2 ntp.nasa.gov www.freebsd.org clock.via.net 64.236.96.

朱雀/ 2022年01月10日 00:29/ 0 赞/ 199 阅读