使用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, 亲测可用)

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