栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

anaconda3环境py37版本+tf1.15+cuda10+cudnn7.6

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

anaconda3环境py37版本+tf1.15+cuda10+cudnn7.6

win10下原安装了py37+tf1.13+cuda10+cudnn7.6,用于tf1框架检测的运算,在C盘系统占空间太大,想移到其他盘,谁知怎么安装都出问题,GPU 不运行,而且训练的模型都保存在C盘昨时文件夹中,也不方便。

ubuntu系统安装了anaconda3,py3.8,cuda11.4,470驱动。anaconda可支持多个版本的python,故另建一个py37版本,用于tf1 目标检测模型的训练,tf1模型pb文件应用方便。

conda create -n py37 python=3.7

建立名为py37的环境。自动安装cuda10.0及cudnn7.6.5

启动激活conda activate py37

安装tensorflow-gpu==1.15.0,安装API

 运行训练:python object_detection/model_main.py --pipeline_config_path=**/**.config --model_dir=** --alsologtostderr

 该错误主要是gpu内存不足引起,修改model_main.py

import tensorflow.compat.v1 as tf

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

config = tf.ConfigProto()
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

后可进行训练。

退出:conda deactivate

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/286200.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号