linux下安裝tensorflow簡潔教程
-
安裝CUDA和cuDNN()
cat /usr/local/cuda/version.txt檢視CUDA版本
需要將cuda加入到環境變數
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
然後啟用source ~/.bashrc
這個時候nvcc –V就可以檢視CUDA版本了。
2.安裝anaconda
新建資料夾用於放置安裝包 mkdir packages
注:下載連結(到清華源找到5.2版本點選右鍵複製連結地址即可)安裝5.2版本。不用建立虛擬環境直接安裝tensorflow。
sh Anaconda3-5.3.0-Linux-x86_64.sh 安裝檔案
3.安裝tensorflow
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu
清華源
注:如果沒有自動安裝對應的3.6版本。
則先建立tensorflow環境conda create –n tensorflow python=3.6
啟用 conda activate tensorflow
再安裝即可.
測試是否安裝成功。
python
import tensorflow (以後每次都需要先進去tensorflow環境)