tensorflow-GPU加速-win10(NVIDIA GeForce 940MX)下的安装教程(VS2015+Cuda9.0)

本文详细介绍如何在本地环境中安装并配置TensorFlow,包括选择合适的Anaconda Python版本、安装CUDA驱动程序、设置cuDNN环境及通过pip安装TensorFlow等步骤。

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

1.安装Anaconda

由于xx(编译器什么的,没看清楚)原因,完整的tensorflow只支持python3.5,(使用Docker的情况除外)。所以Anaconda一般下载python3.5版本。要添加路径path!我用3.6版本

更换python版本型号的方法:

https://jingyan.baidu.com/album/22a299b5e6e4909e18376a4b.html?picindex=2

 

2.安装CUDA Drivers

资源在这里,选对自己电脑对应的系统即可。下载完以后点开安装,跟着安装指导一步步走下去,最好选择默认路径C盘,可以省去不必要的麻烦。下载1.2G那个

默认是最新的版本(目前2018-4是9.1的版本),改旧版本!

我需要9.0,1.4G那个,下载特别快;当然下载那个十几MB的也行,只不过安装的时候会在线下载

因为:后期测试程序出现:

 

3.下载cuDNN - CUDA for Deep Neural Networks

这其实就是CUDA的深度神经网络支持,下载下来以后解压出来一个cuda文件夹,放到你想放的位置,然后把这个cuda文件夹下的bin目录地址放到PATH里。(要先登陆)一定要与Cuda的版本相对应!!!!

4.通过Pip安装tensorflow

# GPU版本:pip install --upgrade tensorflow-gpu

# CPU版本:pip install --upgrade tensorflow

有点慢

 

5、用一个简单的矩阵乘法测试一下,

import tensorflow as tf

a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

参考于:https://blog.youkuaiyun.com/u010099080/article/details/53418159

https://blog.youkuaiyun.com/joker316701882/article/details/54426004

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值