1、TypeError: new() got an unexpected keyword argument 'serialized_options'
更新 protobuf
pip install -U protobuf
2、AttributeError: module 'tensorflow' has no attribute 'gfile'
tensorflow 包版本问题,版本要求 1.15 >= tensorflow >=1.10,更新 tensorflow 版本
pip install -U tensorflow==1.10 -i https://mirrors.aliyun.com/pypi/simple
3、ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
CUDA 版本要求 9.0,下载对应版本安装
CUDA 下载地址:https://developer.nvidia.com/cuda-toolkit-archive
同时 cuDNN 也需要下载和 CUDA 对应的版本,将解压后的文件拷贝到 cuda 的安装目录(默认/usr/local/cuda-9.0)的对应目录下即可;
cuDNN 下载地址:https://developer.nvidia.com/rdp/cudnn-archive
tensorflow-gpu 运行环境安装检测程序
# -*- coding:utf-8 -*-
import tensorflow as tf
with tf.device('/cpu:0'):
a = tf.constant ([1.0, 2.0, 3.0], shape=[3], name='a')
b = tf.constant ([1.0, 2.0, 3.0], shape=[3], name='b')
with tf.device('/gpu:1'):
c = a + b
sess = tf.Session(config=tf.ConfigProto (allow_soft_placement=True, log_device_placement=True))
# sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(tf.global_variables_initializer())
print (sess.run(c))
欢迎来到这里!
我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。
注册 关于