-
当前Tensorflow-gpu在windows环境下仅支持python3.5,因此需要先安装python3.5(建议使用anaconda创建新环境)
如果已安装anaconda,则只需在命令行中输入以下命令即可:
conda create -n py35 python=3.5 ipykernel
ipython kernel install --user
-
安装CUDA8.0(从Nvidia官网下载后直接按提示安装即可)
-
安装cudnn(直接解压后拷贝至CUDA8.0的目录下)
-
安装tensorflow-gpu
切到py35环境后,执行下列命令:pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl
-
安装完成后,可执行以下代码进行测试:
import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)
在Windows下安装Tensorflow-GPU
最新推荐文章于 2024-10-02 18:13:53 发布