入门机器学习,折腾了大半天,终于把TensorFlow gpu版装好了。。。
网上的教程有些版本都比较老了,也不是特别完整。在这里把自己的环境搭建及安装过程分享以下。
我是在win10下进行搭建的,刚开始想用anaconda搭建,但是被坑了大半天,还是没装好,各种错误在这里不多说了,感觉Win下还是不适合用控制台进行这些安装。想试试的小伙伴可以参考这篇教程https://www.cnblogs.com/HongjianChen/p/8385547.html
于是直接用Pycharm进行安装,pycharm直接去官网下载最新版本http://www.jetbrains.com/pycharm/?fromMenu
有社区版和专业版可以选择,我是下了专业版后,之后用激活码直接激活,安装过程基本是一路NEXT,激活码点这里https://blog.csdn.net/weixin_43641304/article/details/87071864
下载完pycharm后,必须要有python解释器,不然pycharm只是一副没有灵魂的驱壳。直接到python官网进行下载https://www.python.org/我下载的是最新版的3.7。之后在pycharm里需要选择左上角文件-设置-Projiect Interpreter来进行环境配置,选择Add进行添加即可。
在这里说明下python和之后需要安装的cuda及cudnn的版本对应的关系,好多教程没有强调,发布的也比较早,容易有坑。安装cpu版本的可以忽略下面这张图。
可以直接在Nvidia显卡设置里看到能够安装的Cuda版本,我的是10.2,但是我还是下载了10.0,亲测照样能用,只要版本不差太多就好。
之后直接去官网下载cuda的相应版本https://developer.nvidia.com/cuda-downloads
安装过程基本也是一路NEXT,要记住自己cuda的安装路径,因为后面会进行环境变量的设置。建议默认安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA10.0。
安装完成后紧接着必须进行cudnn的安装,网址:https://developer.nvidia.com/rdp/cudnn-download#a-collapse705-91,安装前必须注册, cudnn相当于是cuda这个平台的一个库,直接去官网下载对应版本的cudnn。
选择cuDNN vXXX Library for Windows 10即可。下载完成后是个压缩包,里面有这些文件:
我们需要复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0文件下进行覆盖,注意一定是将文件夹下的dll文件挨个覆盖到里面,一共三个,否则待会儿运行时会报无法找到cudnn64_7.dll的错误。
之后进行环境变量的设置,在桌面上选择我的电脑,右键属性,左侧高级系统设置里的环境变量。
之后将下面这几个变量以及值添加到和PATH同级的系统变量里。
CUDA_PATH = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
CUDA_PATH_V9_0 = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
基本上安装的差不多了,之后在直接在pycharm里进行tensorflow-gpu的安装。打开pycharm,选择左上角文件-设置-Projiect Interpreter里右边的加号,点开后会看到一大堆包什么的,直接搜索tensorflow-gpu进行安装,注意版本的对应,安装过程比较长,需要稍稍等待。
安装完成后,新建一个python项目,输入下面这些代码来测试tensorflow是否成功安装好。
没有报错,安装成功!接下来可以进行各种学习了~~~
-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
填个坑。。。
运行后出现下面这种错误,基本上就是cuda,cdunn,tensorflow的版本没有对应的关系。或者在将cudnn里的dll文件名覆盖到cuda里时直接对文件夹进行操作了,导致软件无法判断。