Tensorflow-GPU环境安装-Windows版
Tensorflow-GPU=CUDA-10.0+ANACODA-3.0+CUDNN-7.6.4
- 进入Anaconda的官网下载,选择自己想下载的平台版本+Python版本,最后点击Download按钮就OK了。下载完成后接下来就是傻瓜式安装,一路确认,想改路径改一下路径就可以。详细图文教程点我。
- 安装好ANACODA后需要确认自己的机子需要什么版本的CUDA和CUDNN,然后在命令行窗口输入命令
pip install tensorflow
pip install tensorflow-gpu
安装tensorflow和tensorflow-gpu
3.安装成功后,去tensorflow的目录下去找D:\software\ANACONDA\Lib\site-packages\tensorflow_core\python\platform
找到一个build_info.py文件 查看自己需要的CUDA和CUDNN版本
4.去CUDA官网下载CUDA软件,笔者是CUDA10.0
点击 Legacy Releases查看历史版本:
选择自己需要的版本:
然后选择windows版本后下载就完事了。
5.傻瓜式安装,一路同意点下去。
6.之后去CUDNN官网下载与CUDA相匹配的CUDNN版本。(需要注册一下)我的cuda10.0对应cudnn 7.6.4。
7.下载完成后将CUDNN解压缩并拷贝CUDNN中的三个文件夹到CUDA文件的路径下。
8.此时还需要将这三个文件的路径添加到环境变量中去。
最后可以测试一下自己是否安装成功:
import tensorflow as tf print(tf.test.is_gpu_available())
输出是True就OK啦!
恭喜你可以开始自己的深度学习从入门到放弃的路程啦。、
作者微信: