(Ubuntu安装六)Ubuntu16.04安装Tensorflow-GPU
参考链接: https://blog.csdn.net/qq_40936141/article/details/81198425#commentBox
一、安装python3.6(如果有此处省略)
添加源并更新命令行:
sudo add-apt-repository ppa:fkrull/deadsnakes
sudo apt-get update
安装指定版本的python:
sudo apt-get install python3.6 python3.6-dev
二、安装并更新pip(如果有此处省略)
安装pip:
sudo apt-get install python3-pip
更新pip:
pip install --upgrade pip
三、用pip安装tensorflow-gpu
我安装的是 tensorflow 1.6.0:
pip install tensorflow-gpu==1.6.0
四、检测报错修改
- (1) 检测报错:
- 解决办法:
pip install h5py==2.8.0rc1
- (2)由于numpy版本问题报错
解决办法:
安装numpy==1.14.0
pip install numpy==1.14.0