当前位置:网站首页>tensorflow 1.14指定gpu运行设置
tensorflow 1.14指定gpu运行设置
2022-07-06 17:10:00 【愚昧之山绝望之谷开悟之坡】
注意:allow_soft_placement=True表明:计算设备可自行选择,如果没有这个参数,会报错。
因为不是所有的操作都可以被放在GPU上,如果强行将无法放在GPU上的操作指定到GPU上,将会报错。
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" # 这里是gpu的序号,指定使用的gpu对象
gpu_options = tf.GPUOptions()
gpu_options.allow_growth = True
config_photo = tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options, log_device_placement=True)
self.sess = tf.Session(config=config_photo)
边栏推荐
- 筑梦数字时代,城链科技战略峰会西安站顺利落幕
- Policy Gradient Methods
- 深度学习之环境配置 jupyter notebook
- Attention SLAM:一種從人類注意中學習的視覺單目SLAM
- Tencent cloud webshell experience
- Mujoco second order simple pendulum modeling and control
- Article management system based on SSM framework
- Mujoco Jacobi - inverse motion - sensor
- 批量获取中国所有行政区域经边界纬度坐标(到县区级别)
- 以机房B级建设标准满足等保2.0三级要求 | 混合云基础设施
猜你喜欢
[C language] dynamic address book
threejs图片变形放大全屏动画js特效
Configuring the stub area of OSPF for Huawei devices
ZYNQ移植uCOSIII
C9高校,博士生一作发Nature!
第七篇,STM32串口通信编程
第五篇,STM32系统定时器和通用定时器编程
New feature of Oracle 19C: automatic DML redirection of ADG, enhanced read-write separation -- ADG_ REDIRECT_ DML
AI超清修复出黄家驹眼里的光、LeCun大佬《深度学习》课程生还报告、绝美画作只需一行代码、AI最新论文 | ShowMeAI资讯日报 #07.06
[Niuke classic question 01] bit operation
随机推荐
Mujoco second order simple pendulum modeling and control
Data sharing of the 835 postgraduate entrance examination of software engineering in Hainan University in 23
一行代码实现地址信息解析
Mongodb client operation (mongorepository)
Basic information of mujoco
C9高校,博士生一作发Nature!
【软件逆向-自动化】逆向工具大全
Use mujoco to simulate Cassie robot
Levels - UE5中的暴雨效果
城联优品入股浩柏国际进军国际资本市场,已完成第一步
Equals() and hashcode()
[software reverse automation] complete collection of reverse tools
Summary of being a microservice R & D Engineer in the past year
Data processing of deep learning
Tencent cloud webshell experience
How do novices get started and learn PostgreSQL?
Zynq transplant ucosiii
Js+svg love diffusion animation JS special effects
Advanced learning of MySQL -- Fundamentals -- four characteristics of transactions
pyflink的安装和测试