Python 是一種功能強大、易于學習和使用的編程語言,它可以用于各種不同的應用場景。在機器學習和計算機視覺中,Python 通過使用顯卡可以更快地運行復雜的算法。
import tensorflow as tf # 檢查當前計算機是否有 GPU if tf.test.gpu_device_name(): print('GPU 已啟用') else: print('沒有可用的 GPU') # 創建一個簡單的 TensorFlow 模型 model = tf.keras.Sequential([ tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)), tf.keras.layers.Dense(10, activation='softmax') ]) # 訓練模型 model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy']) model.fit(x_train, y_train, epochs=5) # 使用 TensorFlow GPU 加速 with tf.device('/gpu:0'): for i in range(100): model.fit(x_train, y_train, epochs=1, batch_size=32, validation_data=(x_val, y_val)) # 保存模型 model.save('my_model.h5')
上面的示例代碼演示了如何使用 TensorFlow 庫創建一個神經網絡,并使用顯卡加速訓練過程。一旦訓練完成,可以將模型保存在一個文件中,以備日后使用。
使用顯卡加速 Python 可以有助于提高運行速度,特別是在計算密集型應用的情況下。雖然并非所有的 Python 庫都支持 GPU 加速,但 TensorFlow 和 PyTorch 等流行的機器學習框架提供了這種能力,可以幫助開發人員更快地構建和訓練復雜的模型。