search

kaishengtai / neuralart

An implementation of the paper 'A Neural Algorithm of Artistic Style'.
MIT License
2.41k stars 353 forks source link

how to specify the GPU card if we have multiple one? #27

Closed SeekPoint closed 7 years ago

kaishengtai commented 7 years ago

You can use cutorch.setDevice(i) (see the README at https://github.com/torch/cutorch) or otherwise set the CUDA_VISIBLE_DEVICES environment variable.