1. 程式人生 > >【已解決】docker + GPU + tensorflow:Tensorflow cannot open libcuda.so.1

【已解決】docker + GPU + tensorflow:Tensorflow cannot open libcuda.so.1

我想在docker中使用GPU + tensorflow + keras + 一些其他的環境。
於是下載了 docker pull ufoym/deepo的映象。
在docker run等命令後import tensorflow出現錯誤,錯誤如題。
查過很多資料,但沒能解決。

最後採取了ufoym/deepo網址的啟動方法,將docker run 改成nvidia-docker run,這樣才可以成功使用tensorflow