以下是主要步骤:
一、Install Anaconda Python
Anaconda download page: https://www.anaconda.com/downloads
二、Built a virtual environment
conda create --name tensorflow(环境名称)
activate tensorflow(打开虚拟环境)
三、使用conda安装tensorflow以及keras
conda install tensorflow-gpu keras-gpu
这里conda会自己安装cudatoolkit以及其他相关的包。
四、测试安装的tensorflow-gpu
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))