Tensorflow 2.1安裝


在windows系統下配置機器學習環境有時是非常頭疼的事情,筆者在安裝TF2.1時也踩了不少坑,這里把安裝過程和安裝中可能遇到的問題羅列出來,也歡迎大家一起來交流。

本文主要包括以下內容:

1.Anaconda python3.7版本安裝

2.Tensorflow 2.1安裝

3.PyCharm安裝

4.Jupyter Notebook安裝

Anaconda 安裝

前往Anaconda官網 Python 3.7 版本,選擇64-Bit Graphical Installer 下載。下載完成后按照引導安裝,采用默認路徑。在Advanced Installation Options步驟,勾選第一個選項框 Add Anaconda to my PATH environment variable,將Anaconda加入環境變量,靜待安裝成功。

Tensorflow 2.1安裝

打開Anaconda Prompt(可在windows左下角搜索框中找到),創建一個TF2.1環境。輸入代碼 conda create -n TF2.1 python=3.7 並運行,選擇y安裝相應軟件包。安裝完成后輸入 conda activate TF2.1 進入TF2.1環境。

如果你的計算機支持英偉達GPU,可安裝cudatoolkit和cudnn包,若不支持則跳過這一步直接安裝Tensorflow 2.1,在不支持英偉達GPU的情況下下載這兩個包后續可能會報錯。在安裝cudatookit和cudnn前請務必更新顯卡驅動。在TF2.1 環境下輸入 conda install cudatoolkit=10.1,選擇y安裝軟件包,輸入conda install cudnn=7.6 ,選擇y繼續安裝。

直接用pip安裝官網的Tensorflow速度會很慢,這里采用清華源鏡像。在TF2.1環境下輸入

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow 

PyCharm安裝

前往PyCharm官網下載 community版本,點擊安裝,選擇默認路徑,將Installation Options中的 Add launchers dir to the PATH選上,左邊三個選項按照自己喜好選擇。安裝完成后,選中Rebort Now重啟電腦。之后創建新項目時,可使用conda environment,TF2.1中的python解釋器。

Jupyter Notebook 安裝

打開Anaconda Prompt,輸入代碼 conda install jupyter notebook 安裝完畢。Jupyter Notebook可在瀏覽器中運行,默認地址 localhost:8888/tree

打開Jupyter Notebook輸入代碼

import tensorflow as tf tf.__version__

輸出'2.1.0'則表示Tensorflow 2.1已經安裝好了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM