windows最简单三行命令安装tensorflow-gpu版(使用anaconda)无需本身安装cuda

中文网上能搜到的tensorflow-gpu安装方法都很是麻烦,须要你看你的显卡,看你的显卡驱动,去nvidia官网本身下载cuda啥的,还不必定跟你最后安装的tensorflow兼容。包括我也被这些辣鸡文章害了,浪费一下午。最后发现直接用conda安装就能自动匹配安装对应的cuda等,还能随意选择版本,很是方便。python

1.安装anaconda

进入官网,拉到最下面,根据你系统是64仍是32位下载安装,通常win10都是64位。安装就按默认选项就行。
image.png网络

2.安装tensorflow

打开Anaconda prompt,而后测试

1.建立tensorflow虚拟环境

conda create -n tensorflow python=3.7spa

2.激活环境

conda activate tensorflowcode

3.安装tensorflow

如下命令默认安装符合你的驱动的最新版tensorflow:
conda install tensorflow-gpu
若是想安装指定版本,好比1.10.0,使用如下命令:
conda install tensorflow-gpu=1.10.0
若是想安装CPU版本,使用如下命令:
conda install tensorflow
若是在安装过程当中出现错误,通常是网络问题,你们懂的,使用ke-xue冲浪get

4.安装完成后测试

首先输入python进入python交互环境,而后输入如下代码:it

import tensorflow as tf 
print(tf.__version__)
print(tf.test.is_gpu_available())

预期结果应该以下图,主要是最后输出True,表示使用了GPU,这样就表示安装成功了。
QQ截图20200717190332.pngio

相关文章
相关标签/搜索