Anaconda
首先在自己的用戶目錄下安裝 anaconda
./Anaconda3-5.0.1-Linux-x86_64.sh
安裝完成之後添加anaconda環境
vim ~/.bashrc
export PATH=/home/zhangyt/anaconda3/bin:$PATH
source ~/.bashrc
添加清華鏡像,執行下面命令兩次,將清華鏡像放在第一位
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
創建pythonn3.6的解釋器
conda create -n py36 python=3.6
添加cuda和cudnn的環境,服務器中已經安裝了cuda和cudnn,需要知道它們的安裝位置
vim ~/.bashrc
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
具體如圖
source ~/.bashrc
先升級一下pip
pip install --upgrade pip
開始安裝tensorflow-gpu=1.13.1
pip install tensorflow-gpu==1.13.1
測試TensorFlow(GPU)是否安裝成功
source activate py36
然後鍵入python
import tensorflow as tf
hello = tf.constant('hello world')
sess=tf.Session()
print(sess.run(hello))
如下顯示,說明tensorflow-gpu已經安裝成功了