為使用Tensorflow的GPU系統加速功能,選擇使用Anaconda虛擬環境建立深度學習環境,相關使用套件版本如下:
Python=3.7
Tensorflow-gpu=2.1.0
keras=2.3.1
安裝過程如下:
conda create -n deeplearning python=3.7
activate deeplearning
conda install jupyter notebook
python -m ipykernel install --user --name deeplearning --display-name”DeepLearning”
conda install tensorflow-gpu==2.1.0
conda insta keras==2.3.1
安裝完成後,開啟Jupyter notebook檢查有無安裝錯誤:
import tensorflow as tf
import keras
print("Tensorflow版本為:",tf.__version__)
print("Keras版本為:",keras.__version__)
print("檢查Tensorflow-GPU是否啟用:",tf.test.is_gpu_available())
如有錯誤再上網找答案。