win10 Anaconda Pycharm安装TensorFlow

最新推荐文章于 2024-06-16 17:58:14 发布

ShawDa

最新推荐文章于 2024-06-16 17:58:14 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

文章标签： python windows cuda tensorflow

本文链接：https://blog.youkuaiyun.com/sinat_36811967/article/details/72973109

本文详细介绍了如何在Windows环境下安装配置CUDA、cuDNN及TensorFlow，并使用Anaconda3进行环境管理，最后通过PyCharm进行开发测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

先安装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'。