search

soumith / imagenet-multiGPU.torch

an imagenet example in torch.
BSD 2-Clause "Simplified" License
401 stars 158 forks source link

multi GPU: select GPU to use and #to use #15

Closed culurciello closed 9 years ago

culurciello commented 9 years ago

in order to use GPUs 3,4 in a system with 4 GPUS, for examples, we would need to change https://github.com/soumith/imagenet-multiGPU.torch/blob/master/models/alexnet_ccn2.lua#L38 to "for i=opt.GPU,nGPU do", right?

culurciello commented 9 years ago

Francisco Massa suggest: just use CUDA_VISIBLE_DEVICES=2,3 before launching your script