用tensorflow,pytorch框架使用GPU,指定GPU问题
发布日期:2021-06-29 11:45:20
浏览次数:4
分类:技术文章
本文共 969 字,大约阅读时间需要 3 分钟。
pytorch:
单个GPU指定使用:
1.在代码上指定(建议这种指定方式)
import osos.environ["CUDA_VISIBLE_DEVICES"] = "2" # 指定第二块gpu
2.使用set_device指定
import torchtorch.cuda.set_device(0) # 不能设定两个,比如0,2
查看pytorch所使用的gpu
torch.cuda.is_available() # cuda是否可用torch.cuda.device_count() # 返回gpu数量torch.cuda.get_device_name(0) # 返回gpu名字,设备索引默认从0开始torch.cuda.current_device() # 返回当前设备索引
tensorflow:
1.指定gpu
import osos.environ["CUDA_VISIBLE_DEVICES"]=‘1’
2.设置定量的GPU使用量
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) #占用现存不超过总现存的0.7
3.按需设置最小显存
gpu_options = tf.GPUOptions(allow_growth=True)sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
查看tensorflow所使用的gpu
import tensorflow as tftf.test.gpu_device_name() # 显示当前使用gpu名字tf.test.is_gpu_available() # true or falsefrom tensorflow.python.client import device_liblocal_device_protos = device_lib.list_local_devices()print(local_device_protos) # 列出所有的本地机器设备
转载地址:https://blog.csdn.net/zz2230633069/article/details/88350454 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
做的很好,不错不错
[***.243.131.199]2024年04月13日 23时55分39秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
电赛 | 19年全国一等奖,北航学子回忆录(下)
2019-04-29
突破!台积电1nm芯片,有了新进展。
2019-04-29
一文读懂全系列树莓派!
2019-04-29
自制一个害羞的口罩,见人就闭嘴,戴着可以喝奶茶
2019-04-29
聊聊我是如何编程入门的
2019-04-29
J-Link该如何升级固件?
2019-04-29
从电子垃圾中提炼黄金,可以!!!
2019-04-29
知乎大神深入解析:单片机晶振脚原理是什么?
2019-04-29
电容有17种?看看详细介绍!
2019-04-29
如何准备电赛?19年电赛经验总结!
2019-04-29
蓝牙:为啥叫“蓝”牙,不叫“白”牙?
2019-04-29
干货 | 如何系统学习 C 语言?
2019-04-29
多层PCB内部长啥样? 3D大图解析高端PCB板的设计工艺
2019-04-29
鸿蒙2.0都来了,快搭个环境玩起来吧!
2019-04-29
PCB散热的10种方法!
2019-04-29
值得收藏!268条PCB layout设计规范
2019-04-29
Keil升级了,Keil Studio 来了!
2019-04-29
关于RS-485总线,这篇很详细
2019-04-29
关于2021年电赛的一些想法,看到就是赚到!
2019-04-29
教你一秒分辨真假芯片!
2019-04-29