真实机下ubuntu18.04安装NVIDIA驱动,CUDA-9.0,CUDNN-7.05,tensorflow过程(三)

安装tensorflow-gpu

安装NVIDIA步骤
CUDA-9.0,CUDNN-7.05步骤
可以参考官方文档的步骤

1.先下载以下软件包:

sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv  # system-wide install

2.创建虚拟环境(推荐)

创建一个新的虚拟环境,方法是选择 Python 解释器并创建一个 ./venv 目录来存放它:

virtualenv --system-site-packages -p python3 ./venv

使用特定于 shell 的命令激活该虚拟环境:

source ./venv/bin/activate  

3. 安装 TensorFlow pip 软件包

方法1:官方方法
在不影响主机系统设置的情况下,在虚拟环境中安装软件包。首先升级 pip:

pip install --upgrade pip

下载(注:国内墙的原因,pip速度亲测是非常非常的慢,过段时间就会出现超时错误,使用该方法未成功)

pip install --upgrade tensorflow

方法2:pip下载引用国内镜像源(强烈推荐)
速度爽到飞起

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.9

方法三:
手动下载安装包并安装

方法四:
下载虚拟环境anaconda,在虚拟环境中下载:

conda install tensorflow-gpu

4.测试

尝试你的第一个 TensorFlow 程序 :

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print sess.run(hello)

成功则出现:Hello, TensorFlow!

安装NVIDIA步骤
CUDA-9.0,CUDNN-7.05步骤

猜你喜欢

转载自blog.csdn.net/sazass/article/details/86755664