七,安装conda install tensorflow-gpu=1.9,cuda9.0 cudnn7.1.2

1,创建虚拟环境

conda create -n tf_gpu_py3 python=3.6

source activate tf_gpu_py3

conda install numpy

conda install matplotlib

conda install scikit-image

2,从指定的源安装

{

这个安装与版本不配套!但是值得借鉴!

https://anaconda.org/anaconda/tensorflow-gpu

conda install -c anaconda tensorflow-gpu (这种安装虽然成功,但是与我cuda9。0的版本是不匹配的,所以不行)

}

conda install tensorflow-gpu=1.9 (cuda9.0 cudnn7_7.1.2)

这个版本自动安装对应的是cuda9.2_0,cudnn的版本是:7.1.2-cuda9.2_0 

我在安装cudnn的时候,选择的是cudnn7.1.4,安装tensorflow-gpu=1.9.0后测试了一下,没出问题!

注意:这里安装的cuda和cudnn是在conda--python环境的,并没有影响其他环境的cudnn,cuda版本。

在我的电脑安装cuda的时候,安装了cuda例子源代码

cd  ~/Documents/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery$

 ./deviceQuery 这行这个程序,会发现系统原来安装的版本没受影响。

3,验证

执行: python

输入:

 import tensorflow as tf
tf.__version__
'1.11.0'

4,安装keras,这里安装这个版本的原因是我要学习的mask-rcnn要求这个版本!

pip install keras==2.0.8

猜你喜欢

转载自blog.csdn.net/yanlizhong62/article/details/84103012