0.保证电脑处于联网状态,笔者电脑配置win10_x64 + gtx 1050 ti,如何查看自己的显卡对应的cuda版本在网上找,此处选择的是cuda9.1+cudnn7.1,如何选择匹配cuda和cudnn的版本在nvidia的网站可以看到,安装文件可以在nividia的网站下载或者其他方法下载。
1.安装包准备
1.1 百度搜索,下载Anaconda3-4.2.0-Windows-x86_64.exe(默认python3.5.2)
1.2 https://github.com/fo40225/tensorflow-windows-wheel找到适合自己cuda和cudnn版本的tensorflow安装包,此处选择的是
tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl 或者使用pip install tensorflow-gpu==1.4.0从网上下载,可以指定版本号,另笔者尝试下,发现也很快。
2.命令行安装
2.1 管理员身份运行: Anaconda Prompt
2.2 查看anaconda的版本: anaconda --version
2.3 python版本选择,查看可用的python版本: conda search --full-name python
2.4 安装指定版本的python(anaconda可以安装多个python环境):
conda env remove -n py36
conda create -n py36 python=3.6
activate py36
deactivate
2.5 安装tesorflow需要的环境:
conda env remove -n tensorflow-gpu
conda create --name tensorflow-gpu python=3.6
2.6 找到下载好的tensorflowxxxx.whl的路径,执行下面的命令:
tensorflow版本不对出错
python -m pip install --upgrade pip
pip install D:/tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl
2.7 验证安装是否成功
python
import tensorflow as tf
hello = tf.constant("hello, Tensorflow!")
sess = tf.Session()
print(sess.run(hello))
致此,安装成功!!!