目前tensorflow还不支持cuda9.1,不小心下成了cuda9.1又不想卸了重下,也不会自己编译成支持cuda9.1的怎么办,看看大神的解决办法 安装环境:
- windows 10 64bit
- python 3.6
安装以下步骤进行安装:
更新GPU驱动—>安装cuda—>安装cuDNN—>安装Tensorflow—>安装keras 1、更新GPU驱动
首先查看机器的GPU型号,查看其是否支持cuda,在Nvidia官网下载对应的最新驱动进行跟新。这一步应该很简单,就不多说了。 2、安装cuda
安装包下载好之后,安装cuda(需要管理员权限),按照安装程序一步一步进行下去即可。安装完成后,在cmd输入nvcc -V查看cuda是否安装成功。 3、安装cuDNN
解压缩下载的cuDNN安装包,得到以下三个文件夹 将其复制在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1中,并在系统环境变量中添加如下项: 其中第一项和第三项是安装cuda时系统自动设置的,我们只需要添加第二项即可。 4、安装tensorflow GPU版本
这一步很简单,在cmd输入 pip install tensorflow-gpu 即可。
输入以下代码,查看是否安装成功 - import tensorflow as tf
- hello = tf.constant('Hello, TensorFlow!')
- sess = tf.Session()
- print(sess.run(hello))
复制代码结果显示: - ImportError: Could not find 'cudart64_90.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 9.0 from this URL: https://developer.nvidia.com/cuda-toolkit
复制代码pip uninstall tensorflow-gpu 重新安装: pip install 文件存放路径\文件名.whl 安装完成后,尝试一下代码: - import tensorflow as tf
- #Creates a 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)
- #Creates a session with log_device_placement set to True.
- sess = tf.InteractiveSession()
- #Runs the op.
- print(sess.run(c))
- from tensorflow.python.client import device_lib as _device_lib
- print(any((x.device_type == 'GPU') for x in _device_lib.list_local_devices()))
- print(_device_lib.list_local_devices())
复制代码如果能顺利通过,并显示如下结果,说明安装成功。 4、安装keras
输入 安装完成后,在python IDE中输入 如果输出 恭喜,keras安装成功,并以tensorflow作为后端。
|