网站建设与设计试题,贵阳建筑公司网站建设,培训制作网站,网站代码需要注意什么一、查看 tensorflow 是否使用了GPU进行计算
import tensorflow as tf
sess tf.Session(configtf.ConfigProto(log_device_placementTrue))
运行程序#xff0c;日志若包含 gpu 信息#xff0c;则使用了 gpu。
二、使用指定GPU
方式一#xff1a;代码#xff1a;
imp…一、查看 tensorflow 是否使用了GPU进行计算
import tensorflow as tf
sess tf.Session(configtf.ConfigProto(log_device_placementTrue))
运行程序日志若包含 gpu 信息则使用了 gpu。
二、使用指定GPU
方式一代码
import os
os.environ[‘CUDA_VISIBLE_DEVICES’]’1’
方式二、命令行
CUDA_VISIBLE_DEVICES’1’ python code.py这个命令是指定程序用哪个 GPU 跑 tensorflow不加的话默认都是显卡 0然后可能就爆了。我们有 n 张显卡可以根据需要指定不同的显卡跑程序的
三、设置GPU使用比例 tf 默认一个程序会占用一整个gpu在程序里手动设置GPU使用比例这样可以一个GPU可以跑多个程序
config tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction 0.9 # 占用GPU90%的显存
session tf.Session(configconfig)三、GPU内存分配
session_conf tf.ConfigProto()
session_conf.gpu_options.allow_growth True # 设置allow_grouth根据运行时的需要来分配GPU内存
sess tf.Session(configsession_conf)
四、查看GPU memory的使用情況 在终端输入命令 nvidia-smi GPUGPU 编号 NameGPU 型号 Persistence-M持续模式的状态。持续模式虽然耗能大但是在新的GPU应用启动时花费的时间更少这里显示的是off的状态 Fan风扇转速从0到100%之间变动 Temp温度单位是摄氏度 Perf性能状态从 P0 到 P12P0 表示最大性能P12 表示状态最小性能即 GPU 未工作时为 P0达到最大工作限度时为 P12。 Pwr:Usage/Cap能耗 Memory Usage显存使用率 Bus-Id涉及GPU总线的东西domaindevice.function Disp.ADisplay Active表示GPU的显示是否初始化 Volatile GPU-Util浮动的GPU利用率 Uncorr. ECCError Correcting Code错误检查与纠正 Compute Mcompute mode计算模式。 Processes 表示每个进程对 GPU 的显存使用率。
输入命令 top 查看动态可用内存