关于tensorflow:windows最简单三行命令安装tensorflowgpu版使用anaconda无需自己安装cuda

30次阅读

共计 697 个字符,预计需要花费 2 分钟才能阅读完成。

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

正文完
 0