您的位置:首页 > 编程语言 > Python开发

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

2018-04-02 17:32 2803 查看
Windows10离线安装tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl(对应GPU版本的tensorflow,对应python3.6)
1. 确保python版本是3.5 64位及以上。(tensorflow从1.2开始支持python 3.6,之前的官方是不支持的)
2. 在安装了Anaconda之后,GPU版本的tensorflow和CPU版本的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即可。或者跑个程序试一下。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
相关文章推荐