当前位置:   article > 正文

Tensorflow-gpu安装教程

tensorflow-gpu安装

1、首先使用conda create -n tensorflow python=3.6(python版本根据需求下载)在Anacoda Prompt下创建一个tensorflow环境

 输入y继续创建

 如上图就是创建成功了。

2、在安装Tensorflow-gpu之前,最重要的就是找到对应版本的cuda和cudnn(如果版本不对应,会导致使用不了),部分版本对应如下图,更多的版本对应可以查询从源代码构建  |  TensorFlow。找到自己须要的版本之后与就可以进行安装了。

 3、首先使用activate tensorflow来进入到之前安装的tensorflow环境,如下图所示就表示进入了tensorflow环境。

 4、CUDA安装,整个cuda的安装包有好几个G,但是我们并不需要全部下载,因为大部分都使用不到,我们真正需要的是cudatoolkit,所以我们直接使用conda install cudatoolkit=10.0安装即可,codatoolkit的版本要严格按照上面的对应版本进行安装。下图即安装完成。

5、cudnn安装,根据上面的对应版本号使用conda install cudnn=7.4安装cudnn,完成之后显示跟上图后三行一样就表示安装成功。

6、安装tensorflow-gpu,这个的安装方式跟cuda和cudnn的安装方式有一点点不一样,cuda和cudnn使用conda进行安装,tensorflow-gpu是使用pip进行下载安装的,还是根据之前的cuda和cudnn的版本对应相应版本的tensorflow-gpu,使用pip install tensorflow-gpu==2.0.0进行安装,下图表示安装成功。

 7、检验是否安装成功,输入python进入python环境,可以看到python版本。

然后输入import tensorflow as tf ,显示成功打开一个dll文件之后,在使用tf.test.is_gpu_available(),查看gpu是否可以使用。如果是True则表示安装成功了,可以使用。GeForce MX350 表示显卡信息。

 到这里tensorflow-gpu就安装完成了。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/95594
推荐阅读
相关标签
  

闽ICP备14008679号