ubuntu16.04+cuda9.0_cudnn7.5+tensorflow-gpu==1.12.0

1、查找可用的tensorflow源,该命令运行后终端会输出所有可用的源

anaconda search -t conda tensorflow

image

2、这里name是上一步中输出源的tensorflow name栏的名称,show命令会在终端输出该源具体的信息和下载需要的指令,执行后如图:

anaconda show name(RMG/tensorflow)

3、卸载

pip uninstall **
conda uninstall **

ubuntu16.04+cuda9.0_cudnn7.5+tensorflow-gpu==1.12.0

在成功装好上面的cuda和cudnn后,执行下面代码就可以获得支持GPU的tensorflow

pip install tensorflow-gpu==1.12.0
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容