1. 程式人生 > >ubuntu下安裝tensorflow-gpu

ubuntu下安裝tensorflow-gpu

首先說明:實驗室的伺服器已經配置好了NVIDIA的環境,所以我只在自己的賬戶下做了如下配置:

1. 首先下載anacon3

wget + 下載連結URL + 想要的版本號
例如:wget + https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ + Anaconda3-5.2.0-Linux-x86_64.sh

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh

2. 更新一下環境變數(在自己的賬戶下)

source ~/.bashrc

3. 安裝tensorflow-gpu

pip install tensorflow-gpu

說明:在python2.7下為pip,在python3.x版本下為pip3。
若在anaconda下統一使用pip即可。