win7下Anaconda安装tensorflow_gpu

win7下Anaconda安装tensorflow_gpu

同步滚动:开

1、 Anaconda的版本:Anaconda3-2020.02-Windows-x86_64 安装步骤略.
2、 查看自己电脑的cuda支持的版本。NVIDIA控制面板

image.png

image.png


如图所示:Cuda驱动版本9.0.176。
cuda版本为:cuda_9.0.176_windowsimage.png

image.png


cudnn版本7.0.5image.png

image.png


cuda和cudnn安装请自行解决
百度云盘:
链接:https://pan.baidu.com/s/1Vfgs4iLCjAlU5Y2swwCJmA
提取码:w4ep

 

3.通过 https://tensorflow.google.cn/install/source#linux
左边菜单可以找到对应的win和mac对应版本
查找tensorflow和cuda的对应版本,我本机的版本对应

image.png

image.png


gpu选择1.5到1.12 python 3.3到3.6 cuDNN =7 CUDA =9
4.打开Anaconda Navigator
image.png

image.png


3.1点击 Environments
3.2点击createimage.png

image.png


3.3image.png

image.png


填写nema 选择python版本 点击create。
创建完成后会出现自己新建的environment。

 

一种方式:命令安装tensorflow_gpu

 

点击image.png

image.png


选择 open Terminalimage.png

image.png

 

出现命令窗口


image.png

image.png

切换国内的镜像(国外下载很慢)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
conda config --set show_channel_urls yes
执行成功后,channels 会出现刚才执行的网址

image.png

image.png

 

  1. 查询可以安装的 gpu版本
    anaconda search -t conda tensorflow-gpu
    我本机上找到


    image.png

    image.png

     

    执行安装命令:
    pip install tensorflow-gpu=1.10.0
    等待安装完成。


    image.png

    image.png


    直接输入python测试或者PyCharm等开发工具测试,(开发工具设置python.exe)
    image.png

    image.png


    测试代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
# 新建一个 graph.
import tensorflow as tf
# 新建一个 graph.
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)
# 新建session with log_device_placement并设置为True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# 运行这个 op.
print(sess.run(c))

image.png

image.png

失败的话,大概率是cuda,cudnn版本问题。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值