autokeras安装踩的坑!

autokeras的安装踩了不少坑,差点把我搞吐了都,记录一下!

1. 首先在autokeras的官网https://autokeras.com/install/,上看见autokeras安装要求:python3,tensorflow >=2.3.0

2. 再到https://pypi.org/project/autokeras/#history,发现autokeras的最新版本为1.0.13,本着用新不用旧的想法,决定安装autokeras-1.0.13

3. 在官网的安装一节中,使用虚拟环境的话,官网推荐使用下面的方式来进行安装

    pip install git+https://github.com/keras-team/keras-tuner.git

    pip install autokeras

得先安装git,使用conda install git进行安装,git安装完成以后再使用上面的两条命令进行安装,其中的过程有点漫长

4. 安装完成以后,运行 tf.test.is_gpu_available() 返回的结果是False,原因是有一个..dll文件找不到,就是cuda版本的问题,我的机器安装的是tensorflow2.4.0对应的cudatoolkit-11.0, cudnn-8.0.5!其实根本原因是在安装的过程中会默认下载安装tensorflow2.5.0,所以如果你的机器的cuda版本不支持tensorflow2.5.0的话,就会有问题!本来想尝试安装tensorflow2.5.0对应的cudatoolkit,但是去tensorflow的官网找了半天也找不到tensorflow2.5.0应该安装哪个版本的cudatoolkit!后来又尝试安装了一个较低版本的autokeras,安装成功,但是在运行官网的示例代码时却报错:AttributeError: 'tuple' object has no attribute 'shape',根本跑不起来!

 

所以,正确的安装姿势是下面的步骤:

1. 打开Anaconda Prompt,输入conda create -n xxx python=3.8,创建一个conda环境,由于我机器安装的python是3.8版本,这里就是python=3.8,xxx是conda环境的名字,我的机器已经事先安装好了tensorflow-gpu==2.4.0,cudatoolkit=11.0,cudnn=8.0.5

2. 输入 conda activate xxx 进入conda环境

3. 直接使用 pip install autokeras==1.0.13 -i https://mirrors.aliyun.com/pypi/simple/ 来安装autokeras的1.0.13版本,这里使用的是阿里云镜像,速度很快,比清华的镜像源快很多,应该是清华那边是校园网的原因;安装的时候还是会默认下载安装2.5.0版本的tensorflow;

4. 运行 pip uninstall tensorflow 将2.0.5版本的tensorflow卸载掉;

5. 运行 pip install tensorflow-gpu==2.4.0 -i https://mirrors.aliyun.com/pypi/simple/ 安装gpu版本的tensorflow2.4.0,安装完成以后运行tf.test.is_gpu_available() 结果返回为True,搞定!!!由于我机器安装的是GPU版本的Tensorflow2.4.0对应的cuda和cudnn,所以这里我安装的是tensorflow-gpu==2.4.0,如果你的机器安装的是其它版本的tensorflow,把版本号改一下就行了,但是官网上的安装要求已经明确了tensorflow的版本必须高于2.3.0.

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值