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