À l'aide de CUDA avec pytorch?

J'ai cherché ici, mais j'ai trouvé seulement dépassée postes.

Je veux courir la formation sur mon GPU. J'ai trouvé sur certains forums que j'ai besoin d'appliquer .cuda() sur ce que je veux utiliser CUDA avec (je l'ai appliqué à tout ce que je pouvais sans faire planter le programme)
Étonnamment, cela rend la formation plus encore.

Puis, j'ai trouvé que vous pouvez utiliser cette torch.set_default_tensor_type('torch.cuda.FloatTensor') d'utiliser CUDA. Avec les deux, rien ne change. Ce qui se passe?

Est-il un moyen fiable pour permettre CUDA sur l'ensemble du modèle?

EDIT: Ceci est signalé comme un doublon. Il n'est pas. Le poste que j'était lié à ne pas répondre à toutes mes questions.

Aussi, ce qui ne MyModel() veux dire? J'ai besoin de plus d'exemples tangibles, comme des exemples de code. (C'est le post, je me réfère à)

InformationsquelleAutor A. Dandelion | 2018-06-20