一、方式一
网络模型、损失函数、数据(包括输入、标注)
找到以上三种变量,调用它们的.cuda(),再返回即可
if torch.cuda.is_available():mynn mynn.cuda()
if torch.cuda.is_available():loss_function loss_function.cuda(…
go version
go 1.20.7 go 1.17 以后都是用 go install 命令 D:\Go\bin\go.exe get -u github.com/nsf/gocode
D:\Go\bin\go.exe get -u golang.org/x/tools/cmd/guru
D:\Go\bin\go.exe get -u github.com/rogpeppe/godef>> Running: D:\Go\bin\go.exe get -u github.com…