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。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容