pytorch训练过程调用tensorflow模型
在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？