tensorflow-gpu : failed to create session

可能需要指定可见GPU

(1) vi ~/.bashrc

(2) 在最后添加:

# added by tensorflow
export CUDA_VISIBLE_DEVICES="0"
(3) source ~/.bashrc

(参考:http://www.cnblogs.com/ZhengPeng7/p/7833352.htm)

猜你喜欢

转载自blog.csdn.net/hhhuua/article/details/80492543
今日推荐