软硬件环境:
硬件:cpu:intel i5 8400显卡:Nvidia 1060
软件:操作系统:win10 1709()版本16299
Anaconda3-5.1.0-Windows-x86_64
cuda_9.0.176_win10
cudnn-9.0-windows10-x64-v7
如果未勾选添加目录,安装完之后要添加目录到路径,否则conda等命令无法识别,如下所示:
来源:QMango的CSDN博客
地址:https://blog.csdn.net/u013211009/article/details/78437098?locationNum=7&fps=1
2.安装tensorflow:
输入如下指令:
3.下载安装cuda:
坑:
析:vs如果不是特别需要就不用安装了,(ps:安装Visual Studio Integration很麻烦,容易踩坑,各种版本不兼容问题)在安装cuda时候选择自定义安装,在cuda子目录下面把Visual Studio Integration勾掉。
注:下载前需要登陆nvidia账号,需要选择对应cuda版本,这里选择对应cuda9.0的版本
下载完解压后,将三个文件夹bin,include,lib复制到cuda安装目录:
5.搭建tensorflow环境:
输入指令(注意这里是py3.6):
6.验证tensorflow:
方法1:
输入python**python环境,然后输入如下指令测试:
import tensorflow as tf
hello =tf.constant('hello,TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
坑:
显示警告:
析:
h5py 和 numpy 版本冲突,h5py 官方已修复合并到master 分支,但是还没发新版,在发版之前可以用降级 numpy 的方法跳过这个问题。降级命令如下:
pipinstall numpy==1.13.0
https://stackoverflow.com/questions/48340392/futurewarning-conversion-of-the-second-argument-of-issubdtype-from-float-to
方法2:
打开spyder后,新建project,然后新建py文件,输入如下代码:
import tensorflow as tf
hello =tf.constant('hello,TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
由此,大功告成!