Win10下 tensorflow gpu版安装

准备:

系统环境: windows10 + Anaconda3 + Pycharm

(1)环境配置:

打开Anaconda Prompt,输入清华仓库镜像,这样更新会快一些: 

输入:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

同样在 Anaconda Prompt 中利用Anaconda创建一个python3.5的环境,环境名称为tensorflow ,输入下面命令:

注意:tf现在已支持在win上安装py3.6版本。详见官网:https://www.tensorflow.org/install/

(3.5环境不是必须的了)

conda create -n tensorflow python=3.5

运行 开始菜单 ->Anaconda3—>Anaconda Navigator,点击左侧的Environments,可以看到tensorflow的环境已经创建好了。

在Anaconda Prompt中启动tensorflow环境:

activate tensorflow

注:当不使用tensorflow时,关闭tensorflow环境,命令为:deactivate(返回默认的python 3.6环境)

(2)安装tensorflow

命令行先激活tensorflow

activate tensorflow

输入命令:pip install tensorflow-gpu   经过漫长的等待

这个时候虽然已经安装好了TensorFlow,但是因为需要GPU加速,我们还需要安装cuda和cuDnn(专门为deep learning准备的加速库)。

(3) 安装cuda,  点我点我  ,安装直接点击next即可(注意安装8.0版Gan2,目前tf最高只支持8.0版,对应的适配cudnn是6.0版)

根据自己的环境选择对应的版本,exe分为网络版和本地版,网络版安装包比较小,执行安装的时候再去下载需要的包;本地版安装包是直接下载完整安装包。这里建议下载本地版

(4)cuDnn库的安装  点我点我,这个比较繁琐,需要填写许多问卷,下载压缩包

下载完cuDnn压缩包后解压,

这样就大功告成了。

下面命令行测试一下tensorflow能否运行,在Anaconda Prompt / cmd命令下进入python环境后,逐行输入:

import tensorflow as tf
hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))

输出为:

b'Hello Tensorflow'

(6)在pycharm中使用tensorflow

新建工程后在  File-Setting–Project Interpreter  选择  tensorflow  下的Python解释器(或者直接在里面搜索 tf-gpu,安装即可,晚会再装cuda+cudnn)

随便跑个HelloWorld测试一下,

import tensorflow as tf

hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))

控制台输出:

b'Hello Tensorflow'

这种方式的好处:不用每次都 开启、关闭环境了(activate tensorflow 、deactivate tensorflow)。

 (7)tensorflow测试next:

①官方教程代码测试:

#Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
#Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
#Runs the op.
print sess.run(c)

输出:

Device mapping:
/job:localhost/replica:0/task:0/gpu:0 -> device: 0, name: Tesla K40c, pci bus
id: 0000:05:00.0
b: /job:localhost/replica:0/task:0/gpu:0
a: /job:localhost/replica:0/task:0/gpu:0
MatMul: /job:localhost/replica:0/task:0/gpu:0
[[ 22.  28.]
 [ 49.  64.]]

②example 测试

详细参考博文  http://blog.csdn.net/sb19931201/article/details/53648615//

 刚刚接触深度学习,还不是怎么了解,先装上利器再说~_~ 欢迎大家有问题指教。

参考:

1. http://m.blog.csdn.net/u012052268/article/details/74202439

2. http://blog.csdn.net/sb19931201/article/details/53648615//

3. http://blog.csdn.net/hdd0411/article/details/71305931?locationNum=8&fps=1

转载于:https://www.cnblogs.com/W-Yentl/p/7783933.html

### 回答1: 要在Win11上安装TensorFlow GPU本,您需要遵循以下步骤: 1. 安装CUDA Toolkit:访问NVIDIA官网下载并安装最新本的CUDA Toolkit。 2. 安装cuDNN:访问NVIDIA官网下载并安装与您的CUDA本兼容的cuDNN。 3. 安装Anaconda:访问Anaconda官网下载并安装最新本的Anaconda。 4. 创建虚拟环境:在Anaconda Prompt中使用以下命令创建一个新的虚拟环境: conda create --name tf-gpu python=3.8 5. 激活虚拟环境:使用以下命令激活虚拟环境: conda activate tf-gpu 6. 安装TensorFlow GPU本:使用以下命令在虚拟环境中安装TensorFlow GPU本: pip install tensorflow-gpu 7. 验证安装:使用以下代码验证TensorFlow GPU本是否已正确安装: import tensorflow as tf print(tf.__version__) print(tf.test.is_gpu_available()) 如果输出的本号为TensorFlow GPU本,并且is_gpu_available()返回True,则表示安装成功。 希望这些步骤能帮助您在Win11上安装TensorFlow GPU本。 ### 回答2: 要在Windows11上安装TensorFlow-GPU本,需要遵循以下步骤: 1. 安装CUDA Toolkit 在安装TensorFlow-GPU之前,需要先安装CUDA Toolkit。CUDA是NVIDIA的并行计算平台和编程模型,它包含用于在GPU上执行数学计算的并行计算库和工具。TensorFlow-GPU需要与CUDA一起使用,以获得GPU加速。 在安装CUDA Toolkit之前,需要检查您的GPU是否支持CUDA,您可以在NVIDIA官网上找到兼容性列表。然后,从NVIDIA官网下载适合您的操作系统的CUDA本。 2. 安装cuDNN cuDNN是CUDA深度神经网络库,它提供了实现深度学习任务所需的GPU加速算法。TensorFlow-GPU需要cuDNN,以获得更快的性能。您可以在NVIDIA官方网站上下载cuDNN。 3. 安装Anaconda Anaconda是一个用于Python数据科学的开源分发,它包含了大量的Python库和工具。将Anaconda安装到Windows11上,可以方便地管理所需的Python环境和库。 从Anaconda官网下载适合您的操作系统的Anaconda本,安装后打开Anaconda Prompt。 4. 创建和激活conda环境 您需要在Anaconda中创建一个虚拟环境来安装TensorFlow-GPU。此外,您可以使用以下命令,激活conda环境: conda create -n tensorflow-gpu python=3.6 activate tensorflow-gpu 5. 安装TensorFlow-GPU 在激活conda环境后,您可以使用以下命令,从pip安装TensorFlow-GPU: pip install tensorflow-gpu 或者,如果您要安装特定本的TensorFlow-GPU,例如2.0.0,请使用以下命令: pip install tensorflow-gpu==2.0.0 6. 测试TensorFlow-GPU 要测试安装TensorFlow-GPU是否正常工作,请使用以下代码,检查其本和能否识别GPU: import tensorflow as tf tf.test.is_gpu_available() 如果输出为True,则表示您已成功安装TensorFlow-GPU,并可以在GPU上运行。 总之,安装TensorFlow-GPU本需要的步骤包括安装CUDA Toolkit、安装cuDNN、安装Anaconda、创建和激活conda环境、最后安装TensorFlow-GPU,最后测试安装。如上操作,需要有一定的计算机基础,操作时需谨慎,确保操作十分准确,否则可能会出现一些不必要的问题。 ### 回答3: 安装Win11 tensorflow-gpu本的方法如下: 1. 安装CUDA和cuDNN: 因为tensorflow-gpu需要依赖CUDA和cuDNN,所以第一步需要先安装CUDA和cuDNN,推荐安装最新本(2021年5月2日的最新本是CUDA 11.3和cuDNN 8.2)。 2. 创建虚拟环境: 在Anaconda Navigator中打开Anaconda Prompt,运行以下命令创建虚拟环境: conda create -n tensorflow-gpu python=3.8 3. 激活虚拟环境: 运行以下命令激活虚拟环境: conda activate tensorflow-gpu 4. 安装tensorflow-gpu: 运行以下命令安装tensorflow-gpu: pip install tensorflow-gpu 5. 验证安装: 运行以下命令验证tensorflow-gpu是否成功安装: python import tensorflow as tf print(tf.__version__) 如果输出的本号是tensorflow-gpu本号,则安装成功。 小提示:安装过程中可能会遇到各种问题,比如CUDA本不匹配、缺少某些库等等,如果不确定如何解决,可以在tensorflow的官网或者GitHub仓库中查看相关的文档或者向tensorflow社区求助。 总结: 以上就是在Win11中安装tensorflow-gpu的方法,希望对大家有所帮助。安装过程需要格外小心,尤其是对于初学者而言,切勿心急冒进。如果遇到问题时,可以参考相关文档或咨询专业人士进行处理。加油!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值