Problem running program on non cuda:0
TungChintao opened this issue · 1 comments
TungChintao commented
When I tried to run the program on a GPU other than cuda: 0, for example, on cuda: 4, the program reported an error：tensors were found at cuda:0 and cuda:4.
setting and result are here
slei109 commented
The default setting for data is on cuda0. You should revise the cuda id for data as well. It can be revised in Common/utils.py to_cuda() function.