前几天电脑坏了,之前弄好的深度学习环境没有了,好久没配都忘了,今天重新配好了,试了不少版本组合,记一下确定能用的软件版本。
废话不多说,建议使用conda命令安装,pip容易出各种毛病
全程直接用conda命令就行,不用其他教程里的去英伟达官网下载这个那个,还配置环境变量这么麻烦了。
会用conda命令的直接看组合:
python版本:3.7.0
cudnn版本:8.2.1
cudatoolkit版本:11.3.1
tensorflow-gpu版本:2.6.0
keras版本:2.6.0
全部直接conda install就行,应该会很顺利,不会出任何问题
装好的环境执行如上代码,输出和我一样说明成功了。
不会用conda命令可以继续看了,不过也很简单
anaconda进入python环境的Terminal,依次执行代码:
conda install cudnn=8.2.1
conda install cudatoolkit=11.3.1
conda install tensorflow-gpu=2.6.0
conda install kears=2.6.0
python版本是创建环境时在anaconda里设置的,注意一下就行。
我提供的版本只是确定能用的,不是最新的,如果以后要装别的东西或者有自己想要的版本要用注意先用search命令先看一下conda支持的channel里有没有你要的软件包
conda search tensorflow-gpu
执行后效果如下
可以看的用conda命令能装的到tensorflow-gpu只有2.6.0,2.5.0。2.5.0再往前太老了没人用了
所以conda的缺点就是各种包支持的版本太少了,你说我非要装一个2.4.0版本的怎么搞?那就用pip命令。pip太杂乱了,出问题了不好排查,很多其他教程给的各种版本经常找不到,pip和conda命令混着用,跟着教程走半天发现推不下去了,全白干。