一、CUDA 9.1
download
查看安装CUDA是否成功,输入cmd指令:
nvcc -V
二、cuDNN7.1.3
download
解压覆盖至.\NVIDIA GPU Computing Toolkit\CUDA\v9.1文件下即可
三、Anaconda3 5.0.1
download
安装过程注意勾选加入到环境变量
四、Anaconda下Tensorflow环境搭建
- 输入cmd指令:
conda create -n tensorflow python=3.6.4 #指定python3.6.4版本
- 激活Tensorflow,输入cmd指令:
activate tensorflow
- 安装.whl文件,输入CMD指令:
pip install --ignore-installed --upgrade tensorflow
#pip install --ignore-installed --upgrade tensorflow-gpu
- 测试Tensorflow
进入编辑环境,输入CMD指令:
python
输入CMD指令:
import tensorflow as tf
hello = tf.constant('hello,TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
显示 b'hello,TensorFlow!'说明Tensorflow配置成功
TensorFlow Models
- 安装GIT download
- 获取models文件
git clone --recurse-submodules https://github.com/tensorflow/models
- 将对应的文件放入\Anaconda3\Lib\site-packages\tensorflow (\Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow) 文件夹下面即可
★折腾来折腾去,我(小白)选择在Anaconda Navigator界面安装
Env
Python 3.6
gdal geotiff keras keras-gpu matplotlib numpy opencv pillow scikit-learn tensorflow tensorflow-gpu tqdm
参考
[1] 蔡铎锋毅. windows10 64位 Tensorflow安装--CUDA 9_1+cuDNN7_1_1+python3_6_4+tf1_6+vs2017. CSDN_blog 2018_03_03.