Win10上基于python3.7的tensorflow-gpu全局环境的搭建

Win10+Python3.7+CUDA10.0+CUDNN7.6.0+Tensorflow-gpu1.14.0

装tensorflow的时候发现网上关于python3.7版本的安装资料比较少,故在此开一贴记录一下安装过程

注意本方法不是在虚拟环境中安装,而是在全局环境中配置。

笔者的显卡是GTX 970M

python3.7安装

具体安装过程这里就不再阐述了,网上都是。安装完后再cmd中输入python3如下:
Win10上基于python3.7的tensorflow-gpu全局环境的搭建_第1张图片

这里有一个管理多个python解释器环境的方法:

找到相应python解释器的安装路径,在里面找到python.exe将其复制到桌面并将其改名为pyhon3.exe然后再将其复制回原位置。以后在cmd中就可以用python3来调用这个解释器了。与该解释器有关的操作可以用python3 -m 这个参数来调用。例如python3 -m pip install tensorflow-gpu在python3的安装路径下安装包。

CUDA10.0安装

https://developer.nvidia.com/cuda-toolkit-archive
下载CUDA10.0下载完后直接一路到底(若选择精简安装,则只需安装第一个即可)

CUDNN7.6.0安装

https://developer.nvidia.com/rdp/cudnn-archive
CUDNN是严格和CUDA版本适配的,这里选择7.6.0 for CUDA10.0。
下载完成后解压复制出里面lib、bin、include三个文件夹到CUDA的默认安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0。(注意这里是合并文件,不是覆盖文件)

Tensorflow-gpu1.14.0安装

在cmd中输入python3 -m pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
这里用清华镜像进行下载。若默认不是下载1.14.0则可以在tensorflow-gpu后加上==1.14.0指定安装

验证安装

Win10上基于python3.7的tensorflow-gpu全局环境的搭建_第2张图片
如需在IDE中使用,例如pycharm,只需设置好安装tensorflow的解释器即可。

你可能感兴趣的:(Win10上基于python3.7的tensorflow-gpu全局环境的搭建)