您的位置:首页 > 其它

tensorflow cpu版本运行,tensorflow-GPU版本安装问题

2019-05-25 16:13 771 查看

tensorflow总是报错一堆问题。都不知道从何下手,查找原因,但是总算在同学的提醒下,百度出一篇文章,解决了问题,文章链接如下:
https://blog.csdn.net/llh_1178/article/details/82260819
其中h5py报错版本过低导致,提升版本:

pip install h5py==2.8.0rc1

(还一度把cr1错写成了crl总是报错,找不到相关版本,然后直接安装2.9.0还出现了系统不支持的错误,最后改成了2.8.0rc1)
第二个问题就是在cpu上运行的,需要在命令行添加如下参数配置:

--clone-on-cpu=True

终于运行成功了
1进行数据转换生成tfrecord文件
所有程序在命令行运行,cmd或者Anaconda prompt终端命令下(每条之间只用一个空格,不是换行,这些是一起运行的)

python download_and_convert_data.py
--dataset_dir=../train
--trainset_name=quiz

2.配置好预训练模型的ckpt文件,再在终端输入如下命令:

python quiz-word-recog/train_image_classifier.py
--dataset_name=quiz
--dataset_dir=./train
--checkpoint_path=./inception_v3.ckpt
--model_name=inception_v3
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits
--train_dir=./train
--learning_rate=0.001
--optimizer=rmsprop
--batch_size=32
--max_number_of_steps=100000#可以用于设置最大的训练步数
--learning_rate_decay_type=fixed \#学习率下降方式固定
--save_interval_secs=300 \
--save_summaries_secs=2 \
--log_every_n_steps=10 #每十步输出一下log看一下过程中参数变化
--weight_decay=0.00004 \
--clone_on_cpu=True

tensorflow-gpu版本安装问题

电脑系统:win10
GPU:GTX1060 6G
按照网上介绍在在CUDA面板上查找到系统下的对应CUDA 是10.0.130,但是下载CUDA10.0和CUDA10.1,没有完全符合的10.0.130,10.0对应的只有10.0.132,总之这两个版本安装都失败,各种卸载CUDA文件,各种添加路径等,都没用!!最后看到人家也是GTX1060用的是CUDA 9.0,于是也试了一下9.0,嗯。这下安装上了,cuDNN选择的是对应的7.0.5版本,安装也没问题,但是安装tensorflow-gpu1.8.0版本的时候,还是无法安装,于是自暴自弃选择了默认版本。嗯,安装上了,但是!运行程序的时候还是出现了一大堆错误,晚上回去再实验吧。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: