首页
编程
java
php
前端
首页
编程
java
php
前端
pytorch中将numpy转换成tensor的问题
将numpy转换成tensor的时候,报错:预期cpu但想要调用gpu,请问有什么解决办法?
预期cpu, 但想要调用gpu, 所以你到底是要.cuda()还是要.cpu()
从numpy创建tensor可以用torch.from_numpy()
cuda或者cpu版本包没安装好吧。
点击展开全文