Closed kotchin closed 5 years ago
The warning shows your GPU 1 is different with your GPU 0. Could you please list your hardware environment?
It says they are different yes, but they aren't. I'm running with 2x RTX 2080 Ti, with the same amount of memory.
I listed above the solution to fix it, which seems to be a problem with Python2 divisions.
I found this fix here.
EDIT: There is nothig wrong with your code, it's a pytorch error. I'm just sharing the fix which seems to have touched other people too.
When executing the step "train psmnet with 4 TITAN X GPUs" there's an annoying warning caused by the necessity of using Python2.7 and pytorch, in full:
I suspect it could prevent the usage of multi-gpu. Luckily, there's an easy fix for it. Go to the file in question causing the warning (torch/nn/parallel/data_parallel.py), and at the top of the file, add the following line:
from __future__ import division
Try again. In case you run into the following issue:
Just install future, e.g. using one of the following commands: