近期帮朋友使用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。