如何在google colab中使用tensorflow 1.6.0,或者如何在colab中使用yolo v3 keras

2024-10-16 20:45:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我试着使用这个github存储库来训练yolo v3 keras模型https://github.com/qqwweee/keras-yolo3,我试着在colab上训练模型,但很快我就明白我需要keras 2.2.4来工作,也需要tensorflow 1.6.0,我无法在colab中下载它们,但很快我就在paperspace(另一个云)中用所需的版本训练了模型,很容易。但它是tensorflow而不是gpu版本,我有一个gpu。我还运行了tf.test。然后我试着安装tensorflow gpu 1.6.0,但在此之前我的意思是说我必须卸载tensorflow(cpu版本)。好的,我用pip卸载完成了,然后运行“pip安装tensorflow gpu==1.6.0”。所有这些都是在paperspace中完成的,我尝试导入tensorflow,但它给我留下了一个错误“ImportError:libcublas.so.9.0:无法打开共享对象文件:没有这样的文件或目录”。在线查看,但修复需要接受中间cuda文件。纸质空间不允许我这么做。如果我运行“pip安装tensorflow”没有问题,tensorflow很容易被导入,但cpu版本不同。我尝试切换到colab并尝试了相同的命令,但出现了一个新错误,pip没有找到“tensorflow gpu 1.6.0”。但是在我读的那篇文章中(https://medium.com/analytics-vidhya/training-yolo-with-keras-85c33cdefe21)共享了一个colab笔记本,我试着运行它,同样的问题。但是他通过他的github共享了它,所以github笔记本也有单元格的输出,并且没有错误。我应该如何解决这个问题


Tags: pip文件https模型版本githubcomgpu