TchaikovskyBear
Results
3
comments of
TchaikovskyBear
I have solved this problem after switching pytorch to version 1.3. Can the program only run in version 1.3?
可以修改一下device_map, 某一层的权重被分配到不同显卡上了, 比如像这样:  这里在vision.transformer.layers.8下就会出现tensor计算不在同一个device上, 像我举得例子里你可以把layer.8整个改在同一个设备上
只能说配环境是DL的一部分了, 说实话这个还不算我经历过的最糟糕的, 好几年前华为的NNIE那套用CAFFE的东西才让我煎熬, 还有早期版本的Ascend