pytorchでGPU2つ以上使う方法
日本語で書いてる記事が少なかったから投稿
multiGPU.py
import torch.nn as nn
#modelの定義した後に
model = nn.DataParallel(model).cuda()
modelにAlexnetなりResnetなりを読み込ませた後nn.DataParallelでラップしてあげるだけでマルチGPU使用可能
More than 5 years have passed since last update.
pytorchでマルチGPU
Last updated at Posted at 2017-10-16
Register as a new user and use Qiita more conveniently
- You get articles that match your needs
- You can efficiently read back useful information
- You can use dark theme
List of users who liked
78