windows gpu tensorflow anaconda

    xiaoxiao2021-03-25  122

    windows系统的从零开始:适合连python都没装的朋友

    平台和软件信息

    windows10python 3.5(目前windows版tensorflow只支持3.5?)anaconda 4.2.0(2,3 用起来也一样)pycharm (方便切换不同版本的python)tensorflow r0.12.1 cuda8.0cudnn5.1

    一、显卡驱动

    检查自己的显卡是否符合要求nvidia官网下载 cuda工具包,根据自己的系统选择相应版本,大概1.2GB,这同时包含了显卡驱动,无需再独立装驱动。下载后是个exe,一步步安装就行。 nvidia官网下载cudnn,简单注册一下并填几个问卷,就可以下载。应下载与cuda对应的版本。下载好后解压到cuda文件夹即可。

    二、安装Anaconda & Pycharm

    官网下载anaconda,下载好后直接安装就好,安装时默认勾选添加到系统路径。个人很喜欢pycharm,因为方便切换不同版本的python。适合既要跑python2程序,又要跑python3程序的。pycharm社区版就够用。无脑安装。

    三、conda中装python3.5虚拟环境

    首先,很关键!添加一个conda镜像,否则简直慢死。。。 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes

    创建虚拟环境 conda create -n py35 python=3.5 anaconda 虚拟环境的意义在于不会影响其他版本python。

    (激活与关闭虚拟环境: activate py35 deactivate py35)

    四、Tensorflow

    在激活的虚拟环境中,安装tensorflow-gpu activate py35 pip install tensorflow-gpu(使用conda指令安装只能装cpu版的) 没有pip的话就conda install pip

    五、 测试

    测试是否装好了tensorflow和cuda:(在虚拟环境中)import tensorflow没有报错,且像下图一样输出 successfully opened CUDA library就对了。 tensorflow.__version__指令可以查看自动装的是哪一版的tensorflow。

    pycharm中跑一些测试程序。 对于习惯用命令行运行程序的人来说,以上已经足够了。对于我这种被matlab惯出来,习惯debug的人来说,用pycharm运行不同版本python程序非常方便。

    在pycharm:file-settings-Project interpreter里面,选择到自己建立的虚拟环境即可,路径应该是安装Anaconda的文件夹\envs\py35\python.exe

    一个小测试程序,看看是不是在用gpu计算

    import tensorflow as tf a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a') b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b') c = tf.matmul(a, b) sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) print sess.run(c)

    输出日志应该显示在用gpu:0之类的。

    跑一个自己的程序感受一下吧!不管速度怎么样,好歹cpu可以继续做别的工作了是不是。

    参考文档

    TensorFlow + CUDA 开发环境配置的坑与丘 conda下载很慢怎么办?Anaconda多环境多版本python配置指导
    转载请注明原文地址: https://ju.6miu.com/read-10633.html

    最新回复(0)