【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" 

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

你可能感兴趣的:(计算机视觉,anaconda,深度学习,pytorch)