tensorflow-GPU docker運行

1.tensorflow安裝版本介紹 在服務器上部署的時候,可能宿主機直接用apt install安裝的CUDA9.1(ubuntu18.04支持apt install 安裝CUDA9.1),此時用pip install tensorflow–gpu 安裝的 1.11.0 不能支持CUDA9.1。所以此時可以採用兩種方案,一種是直接安裝編譯好的 tensorflow-gpu。由於已經在CUDA9.
相關文章
相關標籤/搜索