Я могу запустить свою программу в системе с CUDA 9.2, Quadro M5000 (5) и python 3.7.0 (anaconda), в то время как эта же программа выдает ошибку в другой системе с CUDA 10.0, GeForce GTX 1080 Ti(6) и python 3.7.1
здесь есть ошибка
ForkingPickler(file, protocol).dump(obj)
BrokenPipeError: [Errno 32] Broken pipe
весь код оборачивается в функцию с именем train_valid_model () и просто добавьте главный защитник в сценарий:
def train_valid_model():
# complete code here
if __name__=='__main__':
train_valid_model()
Это работает только тогда, когда num_workers = 0, и если я установлю его на 1 или 2, я получу эту ошибку.Любая идея, почему я получаю эту ошибку, когда программа одна и та же, только версия CUDA и версия Python отличаются?