TensorFlow_GPU在Windows上的安装

踩了一下午的坑,被大佬几句话就给点明白了

我的电脑联想win10.显卡gtx1050,cuda9.0,cudnn7.4.1.5,TensorFlow-GPU1.8.0,anaconda4.4.10(这个应该关系不大)之前主要是安装的这些东西不是配套的,所以如果大家安装失败了可能只是安装的版本不对,不需要改一大堆配置,可能重新安装个别的版本就能成功了,

CUDA9.0 : https://developer.nvidia.com/cuda-90-download-archive

CUDNN:https://developer.nvidia.com/cudnn

tensorflow-gpu:pip install tensorflow-gpu=1.8.0

cudnn下载解压缩后直接放到cuda的目录下和bin、lib、include合并就好,cuda环境变量安装后自动配置成功,不需要改什么

然后就是网上说的

conda create -n tensorflow-gpu python=3.6

activate tensorflow-gpu

pip install tensorflow-gpu=1.8.0

python

import tensorflow as tf

如果这个导入成功了,我感觉基本有戏。

纪念一下运行成功的截图

猜你喜欢

转载自blog.csdn.net/weixin_39454351/article/details/86512441