简介

需要开发Unity AI Agent，可以使用tensorflow进行训练，而Unity只有Win和Mac版本的，所以重新回归Win10系统。。不过，win10也可以使用Linux的子系统了，以后就用win10和linux子系统吧。。。。

这里有Ubuntu16.04的教程：https://blog.csdn.net/qq_35976351/article/details/79325476

安装CUDA-9和CUDNN-7

Anaconda在使用虚拟环境安装的时候，直接自动配置了，这一步可以省略了。

安装Anaconda

Anaconda下载地址：https://www.anaconda.com/download/
选择合适的版本，进行下载安装即可。
记住安装的地址，需要添加环境变量。进入到高级系统设置，然后选择环境变量，在下方的系统环境变量的path中，添加Anaconda的两个环境变量：

貌似CUDA的环境变量自动添加了。。。。。

安装Tensorflow-GPU

打开Anaconda的Prompt命令行窗口，创建Tensorflow的虚拟开发环境：

conda create --name tensorflow_gpu python=3.5

之后，激活环境：

activate tensorflow_gpu

之后安装tensorflow-gpu的组件，顺便安装一个keras的gpu版本

conda install tensorflow-gpu
conda install keras-gpu

之后使用Pycharm作为IDE，进行验证：

import tensorflow as tf

mat1 = tf.constant([[3, 3]])
mat2 = tf.constant([[2], [2]])

product = tf.matmul(mat1, mat2)

with tf.Session() as sess:
    result = sess.run(product)
    print(result)

运行程序，第一次加载的时间比较慢，红色的提醒是我们没有进行编译，可以忽略这些。结果如图：