tensorflow在pycharm运行
时间: 2023-11-25 19:07:06 浏览: 266
使用pycharm新建的demo,使用tensorflow
在PyCharm中运行Tensorflow需要先创建一个Tensorflow虚拟环境,并将PyCharm的解释器设置为该环境。首先,你需要在终端中激活虚拟环境并安装Tensorflow。可以执行以下命令来完成这一步骤:
conda activate tensorflow
pip install tensorflow==1.14.0
安装完成后,你可以将以下代码复制到PyCharm中,并根据需要进行修改:
```python
import tensorflow as tf
X = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
y = tf.constant([[10.0], [20.0]])
class Linear(tf.keras.Model):
def __init__(self):
super().__init__()
self.dense = tf.keras.layers.Dense(
units=1,
activation=None,
kernel_initializer=tf.zeros_initializer(),
bias_initializer=tf.zeros_initializer()
)
def call(self, input):
output = self.dense(input)
return output
model = Linear()
optimizer = tf.keras.optimizers.SGD(learning_rate=0.01)
for i in range(100):
with tf.GradientTape() as tape:
y_pred = model(X) # 调用模型
loss = tf.reduce_mean(tf.square(y_pred - y))
grads = tape.gradient(loss, model.variables) # 使用 model.variables 这一属性直接获得模型中的所有变量
optimizer.apply_gradients(grads_and_vars=zip(grads, model.variables))
if i % 10 == 0:
print(i, loss.numpy())
print(model.variables)
```
然后,你需要将PyCharm的解释器设置为你创建的Tensorflow虚拟环境。在PyCharm的设置(Setting)中,找到Interpreter选项,将解释器更改为虚拟环境中的Python解释器(例如3.7版本)。之后,你应该能够在PyCharm中成功运行Tensorflow代码。
阅读全文