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

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

1.安装anaconda

进入官网,拉到最下面,根据你系统是64还是32位下载安装,一般win10都是64位。安装就按默认选项就行。
windows最简单三行命令安装tensorflow-gpu版(使用anaconda)无需自己安装cuda_第1张图片

2.安装tensorflow

打开Anaconda prompt,然后

1.创建tensorflow虚拟环境

conda create -n tensorflow python=3.7

2.激活环境

conda activate tensorflow

3.安装tensorflow

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

4.安装完成后测试

首先输入python进入python交互环境,然后输入以下代码:

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

预期结果应该如下图,主要是最后输出True,表示使用了GPU,这样就表示安装成功了。
windows最简单三行命令安装tensorflow-gpu版(使用anaconda)无需自己安装cuda_第2张图片

你可能感兴趣的:(tensorflow,人工智能,pycharm,anaconda)