【bug】TypeError: Caught TypeError in replica 1 on device 1.

做三位人体重建的时候，运行时代码报错TypeError: Caught TypeError in replica 1 on device 1.

这个错误的意思是在程序中分配了一个gpu, 但运行时给程序两个gpu，但是其实我只是在配置文件中规定使用gpu：0

原因

因为我的服务器有两张卡，空闲了一张显卡，代码自动调用了它们，然后找了半天错误，最后在代码中将gpu设置成一张就行了。

解决方法

在代码中加入下面两行代码，强制使用一块卡即可

import os 
os.environ['CUDA_VISIBLE_DEVICES'] = "0" 

完美解决！真是太不容易了！