Win10 tensorflow 1.12 gpu + CUDA10 + Anaconda3-4.4 python3.6 安装过程
陆续折腾了几天终于把tensorflow gpu版本装完了。特此记录详细安装过程:(如有什么问题和坑,欢迎交流!^_^)
我装的各版本文件:
Cuda 10
Ananconda3-4.4
tensorflow 1.12.0
Step 1: 安装Cuda和CUDDN
先检查显卡支持的Cuda版本。(本人是1060 6G的):
NVIDIA控制面板->系统信息->组件 查看相应Cuda版本
到官网下载相应的Cuda和CUDDN并安装,以Cuda10为例,安装教程传送门https://blog.****.net/qq_37296487/article/details/83028394
注意:要先安装Visual Studio,才能顺利安装完。安装过程中会提示需要装Visual Studio。
Step 2: 安装Anaconda
网上查到cuda 10 要装tensorflow 1.12.0, 而这个版本tensorflow需要python3.6。所以就要装相应的anaconda
(可以在这里查anaconda和python的相应版本 传送门https://blog.****.net/yuejisuo1948/article/details/81043823)
这里我选择Anaconda3-4.4.0-Windows-x86_64.exe (附:各种anaconda版本下载https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/)
装完Python可以用就算成功了
Step 3:安装tensorflow
官方的轮子用不了(不知道为啥)。后来找到个大神编译好的可以用
下载下来,启动Anaconda Prompt,文件放到Anaconda Prompt运行的目录下(如下图),跑这个命令安装 pip install tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl
装完了检查安装是否成功:
在python里跑
import tensorflow as tf
hello = tf.constant('hello')
sess = tf.Session()
出来这堆包括gpu的信息,说明tensorflow安装成功啦!