TensorFlow学习:环境搭建及安装

    入门机器学习,折腾了大半天,终于把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里时直接对文件夹进行操作了,导致软件无法判断。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值