pytorch训练过程调用tensorflow模型

redredblue

于 2020-09-25 15:52:44 发布

阅读量2.3k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/redredblue/article/details/108797492

在PyTorch训练过程中尝试调用已用TensorFlow训练好的模型时遇到错误。问题可能源于环境中的TensorFlow为GPU版本。在CPU环境下运行仍报错，与交互工具和GLUT初始化有关。解决方法包括在终端运行，解决glutInit错误，通过安装freeglut3和freeglut3-dev包避免OpenGL错误。最终虽能运行，但结果显示为黑色。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在pytorch代码训练里，想要调用tf之前训练好的模型，报错

RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /opt/conda/conda-bld/pytorch_1595629427286/work/aten/src/THC/THCGeneral.cpp:47

虽然我使用cpu调用的tensorflow的checkpoint

with tf.device('/cpu:0'):
sess = tf.Session(graph=self.graph, config=tf.ConfigProto(device_count={'cpu': 0}))
os.environ["CUDA_VISIBLE_DEVICES"] = " "

可能就是一个代码无法涉及两个框架吧

可能是因为环境里安装的tensorflow是gpu版本，最后使用的调用了gpu？

运行coma的latent mode

这个报错是因为交互工具不对，在终端下运行可以。

后来在终端下运行，还

最低0.47元/天解锁文章