при обучении простому коду pytorch коэффициент процессора увеличивается.GPU примерно 0% - PullRequest
Новая
       8

при обучении простому коду pytorch коэффициент процессора увеличивается.GPU примерно 0%

0 голосов
/ 27 января 2019

Я делаю учебник по Pytorch .

Код явно завершен.но у меня есть одна проблема.

Это касается моего коэффициента использования процессора.Если я приступаю к тренировкам, коэффициент использования процессора увеличивается до 100%.но GPU составляет примерно 0%.

Я установил CUDA 9.2 и cudnn.и я уже проверил массаж на предмет torch.cuda.is_available()==True.

это нормально, или мои настройки неверны?

1 Ответ

0 голосов
/ 29 января 2019

1 .. Вы явно загружали свою модель и вводили тензоры в графический процессор, показывая следующее https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#training-on-gpu

Например, 

# Configure your device
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# Upload your model onto GPU
net.to(device)
# Upload your tensor onto GPU
inputs, labels = inputs.to(device), labels.to(device)

2 .. Вы также можете использовать «gpustat» для проверки использования графического процессора. https://github.com/wookayin/gpustat

После установки вы можете набрать "gpustat" на терминале
Если ваш код работает на GPU, его использование увеличится.

3 .. И проверьте, добавили ли вы следующий путь CUDA в файл bashrc.
Следующий путь CUDA - это общий путь в Ubuntu Linux,
но этот путь может отличаться в зависимости от ОС или настроек.

Вы можете открыть файл bashrc, набрав vim ./.bashrc
когда ваш текущий каталог дома, если вы используете Ubuntu Linux. 

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

4 .. Также проверьте, установлен ли графический драйвер набрав nvidia-smi на терминале, если вы используете Ubuntu Linux.

...