配置keras 3的后端使用PyTorch

方法一:修改keras.json文件

如果Keras 3以前使用过其它的后端,导入过Keras,那么在用户目录下会生成keras.json文件。例如,我前面使用过TensorFlow,现在想切换为yTorch。

在这里插入图片描述

当前keras.json文件的内容如下:
在这里插入图片描述

将文件中backend": "tensorflow"修改为"backend": "torch"
在这里插入图片描述

方法二:在python代码中修改环境变量

在python代码中,导入keras之前,使用os.environ['KERAS_BACKEND'] = 'torch'将环境变量KERAS_BACKEND的值设置为torch(备注:设置成功以后,只要后端不变,以后的代码就不需要os.environ['KERAS_BACKEND'] = 'torch'这条语句了):

import os

# 一定要在导入keras之前设置KERAS_BACKEND环境变量
os.environ['KERAS_BACKEND'] = 'torch'

import keras
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值