Tensorflow 建模过程遇到的问题

1. 问题:运行出现errors_impl.InvalidArgumentError: Cannot assign a device for operation,并显示无gpu可以,仅cpu可用

解决方案:在模块导入部分加入此代码,可解决

import ctypes
ctypes.CDLL('libgomp.so.1', mode=ctypes.RTLD_GLOBAL)
ctypes.cdll.LoadLibrary('libcusolver.so.9.0')

2.在运行inception_v3和inception_v4模型时遇到

Negative dimension size caused by subtracting 3 from 2 for 'InceptionV3/InceptionV3/Mixed_7a/Branch_0/Conv2d_1a_3x3/Conv2D' (op: 'Conv2D') with input shapes: [1,2,2,192], [3,3,192,320].

解决方案:将conv2d的VALID改为SAME

3.mac上切换python的运行版本

解决方案:mac默认的python版本时2.7,若想切换为用户自己安装的3.x版本,则

open ~/.bash_profile

在该文件最后添加

alias python="/Library/Frameworks/Python.framework/Versions/3.6/bin/python3.6“

即可

发布了46 篇原创文章 · 获赞 17 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/Jum_Summer/article/details/100085937