报错原因.
- 定义张量时类型为
C
P
U
\rm CPU
CPU,将其
r
e
q
u
i
r
e
s
_
g
r
a
d
\rm requires\_grad
requires_grad 设置为
T
r
u
e
\rm True
True 后执行
.
c
u
d
a
(
)
\rm .cuda()
.cuda() 操作,报如题所示错误。
opt.requires_grad = True
opt = opt.cuda()
解决方案.
- 需要先执行
.
c
u
d
a
(
)
\rm .cuda()
.cuda(),再设置自动微分属性。
opt = opt.cuda()
opt.requires_grad = True