您的位置:首页 > 其它

win10 Anaconda Pycharm安装TensorFlow

2017-06-10 00:19 375 查看
先安装CUDA和Cudnn:

CUDA安装:https://developer.nvidia.com/cuda-downloads

下载合适自己系统版本的文件,我的是Windows x86_64 10 exe(network),按照要求安装好就行了。

Cudnn安装:https://developer.nvidia.com/cudnn

要登录,注册下就好了。注意要下载v5.1版本的,如果用的6.0版本,最后跑例子会出现加载不了模块的问题(ImportError: DLL load failed: 找不到指定的模块)。然后解压到C盘,将cuda和cuda\bin添加到系统环境变量中。我还把cuda三个文件夹的文件分别放到了CUDA(C:\Program Files\NVIDIA GPU
Computing Toolkit\CUDA\v8.0)安装目录的对应文件夹中。

Anaconda3下安装TensorFlow:

先安装Anaconda3:

推荐到这个清华的镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

选择对应的版本,我选的最新的Anaconda3-4.4.0-Windows-x86_64.exe

目前Windows版本的TensorFlow只支持Python3.5以上,在这些镜像里已经默认是Python3.5了。

安装完成之后,打开Anaconda Prompt,输入清华的仓库镜像,更新包更快:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

推荐第一行命令输入两次,以便把这个镜像地址放在首位。

之后安装TensorFlow:

打开Anaconda Prompt,输入:conda
create -n tensorflow python=3.5

安装完以后,输入:activate
tensorflow

激活后,我选择安装的是GPU版本:pip
install --upgrade --ignore-installed tensorflow-gpu

如果选择CPU版本:pip
install --upgrade --ignore-installed tensorflow

关闭:deactivate
tensorflow

Pycharm配置:

我下载的是edu版本:https://www.jetbrains.com/pycharm-edu/download/#section=windows

安装后选择project
interpreter:C:\Users\****\AppData\Local\conda\conda\envs\tensorflow\pyhton.exe

注意把\tensorflow和\tensorflow\Scripts加入环境变量。

最后测试TensorFlow是否安装好:

import tensorflow as tf
hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))

之后会显示电脑的显卡名字和显存等信息,最后输出b'Hello!TensorFlow'。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: