pycharm用GPU跑代码（tensorflow版）

pycharm用GPU跑代码（tensorflow版）import osimport tensorflow as tfconfig = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))sess = tf.compat.v1.Session(config=config)# 选择编号为0或1

贝嘎贝嘎贝嘎处

3912人浏览 · 2021-09-01 12:19:28

贝嘎贝嘎贝嘎处 · 2021-09-01 12:19:28 发布

pycharm用GPU跑代码（tensorflow版）

import os
import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)
# 选择编号为0或1或2的GPU，根据你自己电脑的情况来填
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

PTA团体程序设计天梯赛 L3-033 教科书般的亵渎（DP+状态压缩+剪枝）（30/30) (Java实现)

九条可怜最近在玩一款卡牌游戏。在每一局游戏中，可怜都要使用抽到的卡牌来消灭一些敌人。每一名敌人都有一个初始血量，而当血量降低到 0 及以下的时候，这名敌人就会立即被消灭并从场上消失。现在，可怜面前有 n 个敌人，其中第 i 名敌人的血量是 ai，而可怜手上只有如下两张手牌：如果场上还有敌人，等概率随机选中一个敌人并对它造成一点伤害（即血量减 1），重复 K 次。对所有敌人造成一点伤害，重复该效果