和网上找到的答案都不一样。
问题是因为我在__init__外面的函数中定义了一个卷积操作
然后就一直报错。
具体原因之后再查吧。
win10下查看pyttorch实际使用的cuda版本,安装多个cuda(cuda9.0和10.0),并切换cuda版本
keras实现多gpu训练,单gpu保存模型,单gpu预测