使用conda安裝tensorflow-gpu 2.0

tensorflow1.x時, 使用conda安裝tensorflow-gpu的最大好處是會自動安裝cuda、cudnn等依賴。 現在2.0出來了, 但conda暫時還不支持其安裝, 最新的版本仍是1.14。 若使用pip安裝, 大概率會碰到報錯:dynamic library 'libcudart.so.10.0' not found. 這是因爲對應的cuda環境沒裝好。
可以手動安裝對應的cuda版本, 也可以通過conda取巧:

conda install tensorflow-gpu=1.13
pip install tensorflow-gpu==2.0

首先通過conda安裝1.13版本的tensorflow-gpu, 它會將2.0需要的cudart安裝好。 然後再用pip安裝tensorflow2.0覆蓋conda環境中的tensorflow1.13。
需要注意的一點是, conda安裝的tensorflow最好指定1.13版本, 它的cudart和2.0的相同。 而1.14的cudart則是10.1版本, 與2.0不兼容。 更早的版本則未測試過。

(2019.10.9, 親測可用)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章