RuntimeError: Expected object of backend CPU but got backend CUDA for argument #4 'mat1'

data=data.cuda()

如何在GPU 上运行程序

CUDA_VISIBLE_DEVICES=O,1 python3 main.py //程序只能使用GPU:1
发布了234 篇原创文章 · 获赞 61 · 访问量 12万+

猜你喜欢

转载自blog.csdn.net/weixin_42528089/article/details/104796265
今日推荐