windows下的tensorflow搭建

第一步:

在官网下载CUDA进行安装,在安装之前需要先下在图形驱动程序NVDIA Drives,注意其版本要和自己的显卡的型号相对应。可能出现的错误是“图形驱动程序无法找到兼容的图像硬件”,解决方案参见http://yuanlanxiaup.iteye.com/blog/1330287。之后进行CUDA的安装,安装与官网上的一致。

第二步:

进行CUDNN的安装,CUDNN在官网上可以直接下载,不过需要填写相应的问卷或者注册后才能下载。下载完成后将其解压,对应的文件分别复制到CUDA的文件夹下,(windows下没有lib64,不知道需不需要关联环境变量?)。

第三步:

在清华镜像网站上下载AancondaAnaconda3-4.2.0-Windows-x86_64对应的是自带Python3.5Anaconda3-5.0.1-Windows-x86_64对应的是自带Python3.6。根据需要下载不同的版本。

可能遇到的问题:


解决方案:

 

第四步:

 

按照上述操作:


pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.2.0-cp35-cp35m-win_amd64.whl

在安装GPU版本的tensorflow时总是出现

 

所以在清华镜像网站上下载了tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl版本,使用以下命令安装:

pip install F:/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl

安装成功。

猜你喜欢

转载自blog.csdn.net/qq_30622831/article/details/79574806
今日推荐