tensorflow-gpu2.0版安装成功测试

本文记录了在win10 x64系统上,使用conda成功安装TensorFlow-GPU 2.0的过程。内容包括更新显卡驱动至最新,安装CUDA 10.0和CUDNN,配置环境变量,最后在PyCharm中验证安装成功。
摘要由CSDN通过智能技术生成

win10,tensorflow-gpu2.0版安装成功测试

最近在conda中安装了tf2.0gpu版,环境为win10x64,Nvidia GT940M。

按照tf官方要求:

1更新了显卡驱动;
2安装了cuda10.0(注意tf2.0只支持cuda10.0);
3安装了cudnn;
4配置环境变量。

一切就绪后,在pycharm中更换python解释器,运行如下代码:

注:tf.Session在2.0版本中取消了。

```python
import tensorflow as tf

a = [1.0,2.0,3.0,4.0]
b = [[0.3,0.1],[0.0,0.2]]

sa = tf.nn.softmax(a)
sb = tf.nn.softmax(b)
print(sa)
print(sb.numpy())

运行成功后有如下显示:

```python
C:\Users\yangge\Miniconda3\envs\nlpenvgpu\python.exe C:/PythonPrograms/seq2seq/gputest.py
2019-10-10 07:28:32.981947: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值