使用conda快速在windows下安装GPU版本的Tensorflow+CUDA+cudnn

  • Post author:
  • Post category:其他


之前我尝试过很多种方法,首先要找python对应版本的tf,又要找tf对应版本的cuda,最后还要找cuda对应版本的cudnn,麻烦事一大堆,还不一定能一次安排上,偶然间发现了conda的强大,来跟大家分享一下。

首先推荐安装Anaconda,有多好用就不多说了,基本上有常用的各种包,关键是还有大杀器conda,从此以后就告别pip了。

首先,确认你的python版本是3.6(反正不能是3.7),如果要快速迁移python版本,直接在命令行下输入:

conda install python==3.6

然后直接:

conda install tensorflow-gpu

conda会自动帮你下载gpu版本的tensorflow所需的所有环境,包括cuda和cudnn,我忘了截图,印象中tf目前最新版本1.12.0对应的是cuda9.0.1和cudnn7.1.4,但是这都不用自己操心,conda自动识别下载,默默等待下载安装完成就好了,亲测可用。



版权声明:本文为Setsuna_Ogiso原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。