win10_x64 - tensorflow_gpu cuda9.1+cudnn7.1 - anaconda环境搭建

0.保证电脑处于联网状态,笔者电脑配置win10_x64 + gtx 1050 ti,如何查看自己的显卡对应的cuda版本在网上找,此处选择的是cuda9.1+cudnn7.1,如何选择匹配cuda和cudnn的版本在nvidia的网站可以看到,安装文件可以在nividia的网站下载或者其他方法下载。

1.安装包准备

1.1 百度搜索,下载Anaconda3-4.2.0-Windows-x86_64.exe(默认python3.5.2)

1.2 https://github.com/fo40225/tensorflow-windows-wheel找到适合自己cuda和cudnn版本的tensorflow安装包,此处选择的是

tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl 或者使用pip install tensorflow-gpu==1.4.0从网上下载,可以指定版本号,另笔者尝试下,发现也很快。

2.命令行安装

2.1 管理员身份运行: Anaconda Prompt

2.2 查看anaconda的版本: anaconda --version

2.3 python版本选择,查看可用的python版本: conda search --full-name python

2.4 安装指定版本的python(anaconda可以安装多个python环境):

conda env remove -n py36

conda create -n py36 python=3.6

activate py36

deactivate

2.5 安装tesorflow需要的环境:

conda env remove -n tensorflow-gpu

conda create --name tensorflow-gpu python=3.6

2.6 找到下载好的tensorflowxxxx.whl的路径,执行下面的命令:

tensorflow版本不对出错

python -m pip install --upgrade pip

pip install D:/tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl

2.7 验证安装是否成功

python

import tensorflow as tf

hello = tf.constant("hello, Tensorflow!")

sess = tf.Session()

print(sess.run(hello))

致此,安装成功!!!

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值