一、環境要求
1、操作系統:Windows 7或更高版本,64位操作系統
2、Python:3.5、3.6、3.7、3.8(64位),注意需要添加到環境變數PATH中
3、CUDA:參考官網的要求安裝對應版本
4、cuDNN:參考官網的要求安裝對應版本
二、使用pip安裝TensorFlow
1、使用pip命令在命令行中安裝TensorFlow,打開命令行窗口,輸入以下命令:
pip install tensorflow
2、如果之前已經安裝過TensorFlow,還可以更新到最新版本:
pip install --upgrade tensorflow
三、使用Anaconda安裝TensorFlow
1、下載Anaconda,安裝完成後打開Anaconda Prompt。
2、創建新的虛擬環境,例如名為「myenv」:
conda create -n myenv python=3.7
3、激活新的虛擬環境:
conda activate myenv
4、安裝TensorFlow:
pip install tensorflow
5、運行TensorFlow:
python import tensorflow as tf
註:如果需要使用GPU,還需要安裝CUDA和cuDNN並配置好環境變數。
四、安裝TensorFlow-gpu
1、首先需要安裝CUDA和cuDNN,並配置好環境變數。
2、使用pip安裝TensorFlow-gpu:
pip install tensorflow-gpu
3、運行TensorFlow-gpu:
python import tensorflow as tf
註:如果安裝TensorFlow-gpu時遇到問題,可以嘗試升級pip版本,也可以在官網下載.whl文件進行安裝。
五、安裝TensorFlow的其他版本
1、安裝特定版本的TensorFlow:
pip install tensorflow==2.3.1
2、安裝TensorFlow的預發布版本:
pip install tf-nightly
3、安裝廢棄的TensorFlow 1.x版本:
pip install tensorflow==1.15
六、常見問題解決辦法
1、如果PIP安裝速度很慢,可以使用清華鏡像源進行加速。在命令行中輸入以下命令:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
2、如果pip安裝TensorFlow時出現「Microsoft Visual C++ 14.0 is required」的錯誤,可以安裝Microsoft Visual C++ Build Tools:
pip install --upgrade setuptools pip install --upgrade pip pip install tensorflow
3、如果CUDA或cuDNN的版本不匹配,會出現導入TensorFlow時的錯誤,需要重新安裝對應版本的CUDA和cuDNN。
4、如果出現導入TensorFlow時「could not load dynamic library」的錯誤,可以將對應的.dll文件複製到TensorFlow所在的文件夾中。
七、結論
隨著TensorFlow的使用越來越廣泛,安裝TensorFlow也變得越來越簡單。通過本文介紹的安裝方法,相信大家已經可以更輕鬆地在Windows系統上使用TensorFlow了。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-tw/n/192523.html