2021-03-08|閱讀時間 ‧ 約 2 分鐘

環境設定

    為使用Tensorflow的GPU系統加速功能,選擇使用Anaconda虛擬環境建立深度學習環境,相關使用套件版本如下:
    Python=3.7 Tensorflow-gpu=2.1.0 keras=2.3.1
    TensorFlow對應Keras版本
    TensorFlow對應Keras版本
    TensorFlow對應CUDA版本
    安裝過程如下:
    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())
    如有錯誤再上網找答案。
    分享至
    成為作者繼續創作的動力吧!
    © 2024 vocus All rights reserved.