tensorflow-gpu的使用

近期帮朋友使用tensorflow训练了一套模型，使用的是tensorflow。 因为素材比较大的问题，所以这里特地的使用了gpu训练，也是做了一些自己没做过的东西。

1.安装环境

    电脑环境：windows10

    python版本: 3.7.0

    gpu配置: NVDIA GeForce GTX 750

2.安装过程

首先呢，创建一个虚拟环境，我这边使用的是anaconda

#创建虚拟环境

conda create -n tensorflowEnv python=3.7.0

# 进入虚拟环境

activate tensorflowEnv

# 安装 tensorflow-directml-plugin 这是一个windows平台使用的Gpu插件

pip install tensorflow-directml-plugin -i https://pypi.tuna.tsinghua.edu.cn/simple

# 接下来输入 在python程序中检测一下gpu是否可用即可。

python

import tensorflow as tf

tf.config.list_physical_devices

3 测试结果

4 遇到的错误

            ERROR: Cannot uninstall ‘wrapt‘. It is a distutils installed project and thus we cannot accurat

这个错误是因为wrapt包是dbUtils使用的，而tensorflow包，是需要使用到指定版本的wrapt的，所以这里需要卸载重装的原因。只需要在安装tensorflow排除这个包即可。

ImportError: urllib3 v2.0 only supports OpenSSL 1.1.1+, currently the 'ssl' module is compiled with 'OpenSSL 1.1.0h 27 Mar 2018'. See: https://github.com/urllib3/urllib3/issues/2168

这个错误是因为说，我这边anaconda创建的python环境自带的基础包版本不匹配的原因，这边只需要冲寻装一下urllib3即可。 我这边指定版本为1.15.0。