Windows10下安装原生TensorFlow GPU版
(下载cuDNN需要先注册NVIDIA开发账户并登录才能看到下载界面)
CUDA 9 要TF 1.5版本才支持
- 安装CUDA
- 解压cuDNN到一个你喜欢的位置,复制文件夹的绝对路径并加到 PATH 环境变量去,然后把该文件夹下的bin文件夹的路径也放到PATH里去
- 安装Anaconda Python 3.5 version:
注意windows tf 仅支持 python 3.5
- 安装 tf
为了避免之后使用 pip 安装 tf 报错,请先在cmd运行以下代码 pip install --ignore-installed --upgrade pip setuptools
之后安装tfpip install tensorflow-gpu==1.1.0
接来下检验安装是否成功
import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))
另外,如果用conda的话一定要注意python版本,目前最高支持3.5
conda create --name tensorflow python=3.5 activate tensorflow pip install tensorflow-gpu
原文作者: filwaline
相关推荐
JM 2020-06-21
Micusd 2020-11-19
xjtukuixing 2020-10-27
lybbb 2020-10-15
lybbb 2020-09-29
ghjk0 2020-09-24
yamaxifeng 2020-09-09
GDGYZL 2020-08-28
lybbb 2020-08-28
Icevivian 2020-08-25
comwayLi 2020-08-16
carbon0 2020-08-16
源式羽语 2020-08-09
sherry颖 2020-08-01
songbinxu 2020-07-19
sherry颖 2020-07-18
Niteowl 2020-07-15
Kindle君 2020-07-15
源式羽语 2020-07-04