choice of GPU - Githubissues

DmitryUlyanov / texture_nets

Code for "Texture Networks: Feed-forward Synthesis of Textures and Stylized Images" paper.

Apache License 2.0

1.22k stars 217 forks source link

choice of GPU #38

Closed SantaFlamel closed 8 years ago

SantaFlamel commented 8 years ago

Hi! I didn't manage to find in your code an option to choose the number of graphics card. If there is such an option, how to use it? Thank you in advance. Sincerely

DmitryUlyanov commented 8 years ago

You can use CUDA_VISIBLE_DEVICES=<id> th train.lua ... construction. This will select id device for computations.