Windows10离线安装tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl(对应GPU版本的tensorflow,对应python3.6)

Windows10离线安装tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl(对应GPU版本的tensorflow,对应python3.6

1. 确保python版本是3.5 64位及以上。(tensorflow1.2开始支持python 3.6,之前的官方是不支持的

2. 在安装了Anaconda之后,GPU版本的tensorflowCPU版本的tensorflow安装相同,在以上的基础上(缺少的文件安装完毕),进入Anaconda3\Lib\site-packages\tensorflow中,执行pip install tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl 即可。注,如果要卸载此版本的tensorflow,则在Anaconda3\Lib\site-packages\tensorflow路径中,pip uninstall tensorflow.

3. 安装CUDA(先确保显卡支持CUDA)

1) TensorFlow 1.6对应CUDA 9.0去官网下载对应版本的安装包(~1.4G)。双击下载的exe文件进行安装。

2) 添加环境变量,使具有一下4个环境变量

 


3) 测试一下CUDA是否安装成功:

Cmd(命令行)输入nvcc -V,看到版本信息就表示安装成功了:


4. 安装cuDNN

TensorFlow1.6对应的版本是cuDNN7.0,去官网下载对应的安装包(其实就是个压缩包),将里面的文件解压,然后分别添加到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0对应的三个文件夹中

5. 测试安装成功

cmd中输入import tensorflow即可。或者跑个程序试一下。

猜你喜欢

转载自blog.csdn.net/qq_19307465/article/details/79792400