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

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

anaconda search -t conda tensorflow

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

猜你喜欢

转载自www.cnblogs.com/BambooEatPanda/p/11360466.html