tensorflow-GPU加速-win10(NVIDIA GeForce 940MX)下的安装教程

1.安装Anaconda

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

2.安装CUDA Drivers

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

3.下载cuDNN - CUDA for Deep Neural Networks

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

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.csdn.net/u010099080/article/details/53418159

https://blog.csdn.net/joker316701882/article/details/54426004


猜你喜欢

转载自blog.csdn.net/lusics/article/details/80033051