diff --git a/train.lua b/train.lua index 69752d6..da4a98a 100644 --- a/train.lua +++ b/train.lua @@ -951,6 +951,8 @@ function main() print('using CUDA on GPU ' .. opt.gpuid .. '...') if opt.gpuid2 >= 0 then print('using CUDA on second GPU ' .. opt.gpuid2 .. '...') + -- Disable cross-GPU checks + cutorch.setKernelPeerToPeerAccess(true) end require 'cutorch' require 'cunn'