+-
中文网上能搜到的tensorflow-gpu安装方法都非常麻烦,需要你看你的显卡,看你的显卡驱动,去nvidia官网自己下载cuda啥的,还不一定跟你最后安装的tensorflow兼容。包括我也被这些辣鸡文章害了,浪费一下午。最后发现直接用conda安装就能自动匹配安装对应的cuda等,还能随意选择版本,非常方便。
1.安装anaconda
进入官网,拉到最下面,根据你系统是64还是32位下载安装,一般win10都是64位。安装就按默认选项就行。
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,这样就表示安装成功了。