我能否这样理解:在使用paddle.to_tensor()
创建tensor时,要对参数place=CUDAPlace(0)
指定;而在对于一般的e.g. paddle.zeros()
之类的api,我只能使用paddle.set_device(string)
来全局设定。而对于一般的tensor,我要改变,又只能用tensor.cuda(0)
才能改。
所以说,对于修改设备,我分别得用三种数据类型paddle.place
,string
(e.g. "gpu:0") ,and int
(e.g. device_id=0)。这明显没有torch一个device方便啊,还是说有统一的方法?
2条答案
按热度按时间ogsagwnx1#
您好,我们已经收到了您的问题,会安排技术人员尽快解答您的问题,请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时,您也可以通过查看官网API文档、常见问题、历史Issue、AI社区来寻求解答。祝您生活愉快~
Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the API,FAQ,Github Issue and AI community to get the answer.Have a nice day!
o2rvlv0m2#
谢谢,我们会作为一个需求向PM反馈的哈。