中文网上能搜到的tensorflow-gpu安装方法都很是麻烦,须要你看你的显卡,看你的显卡驱动,去nvidia官网本身下载cuda啥的,还不必定跟你最后安装的tensorflow兼容。包括我也被这些辣鸡文章害了,浪费一下午。最后发现直接用conda安装就能自动匹配安装对应的cuda等,还能随意选择版本,很是方便。python
进入官网,拉到最下面,根据你系统是64仍是32位下载安装,通常win10都是64位。安装就按默认选项就行。网络
打开Anaconda prompt,而后测试
conda create -n tensorflow python=3.7
spa
conda activate tensorflow
code
如下命令默认安装符合你的驱动的最新版tensorflow:conda install tensorflow-gpu
若是想安装指定版本,好比1.10.0,使用如下命令:conda install tensorflow-gpu=1.10.0
若是想安装CPU版本,使用如下命令:conda install tensorflow
若是在安装过程当中出现错误,通常是网络问题,你们懂的,使用ke-xue冲浪get
首先输入python
进入python交互环境,而后输入如下代码:it
import tensorflow as tf print(tf.__version__) print(tf.test.is_gpu_available())
预期结果应该以下图,主要是最后输出True,表示使用了GPU,这样就表示安装成功了。io