成功解決tensorflow2.0在VS-Code代碼自動提示功能

目前tensorfow2.0穩定版還不能自動提示,網上說是tensorflow懶惰加載依賴引起的,先不管這些了。使用tensorflow-2.0.0-beta(測試版),其中的API跟穩定版基本一致,應該可以滿足大部分需求,如果不行的話,可以將編寫的代碼放到tensorflow-2.0穩定版本運行,這也是一個不錯的選擇。不多說了,附上具體解決方法。

1、pip install tensorflow-gpu==2.0.0-beta,官網說不加gpu默認加了gpu,但是我這裏測試,不加gpu的話,不會去加載gpu跑程序,而且我看帶gpu的安裝包比不帶的要大的多

2、在VS-Code的setting.json中加上 “python.analysis.downloadChannel”: "beta"
setting.json在VS-Code設置中可以查找並編輯。

附上tensorflow2.0官網代碼,可以看出keras成爲tf下的一個包了。

from __future__ import absolute_import,division,print_function,unicode_literals
import numpy as np
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train,y_train),(x_test,y_test) = mnist.load_data()
x_train,x_test = x_train/255.0,x_test/255.0
model = tf.keras.models.Sequential([
    tf.keras.layers.Flatten(input_shape=(28,28)),
    tf.keras.layers.Dense(128,activation='relu'),
    tf.keras.layers.Dropout(0.2),
    tf.keras.layers.Dense(10,activation='softmax')
])
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'])
model.fit(x_train,y_train,batch_size=32,epochs=50,validation_data=(x_test,y_test))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章