中文网上能搜到的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,这样就示意装置胜利了。