How to use the device function (pytorch)

・In Japanese
＜premise knowledge＞
・Python

■Description
Specify the GPU (cuda) to use with pytorch. cuda is a development environment for GPU provided by NVIDIA.

■Example
First, check if your PC has a GPU that can be used with pytorch.

Next, store either "cuda" or "cpu" in a variable as a device name that can be used, as follows.

By using ".to", you can perform operations using cuda or cpu as follows.

If you get an error message like the one below, the GPU is set to be used even though it cannot be used. Let's check if the formula in #(1) above is set correctly.

　"Torch not compiled with CUDA enabled"

List of related articles TOP MENU Python library ・pip ・MeCab Common ・class ・pickle ・read/readline numpy ・asfarray ・digitize ・expit ・linalg.solve ・meshgrid ・mgrid ・ndmin ・pad ・poly1d ・polyfit ・prod ・shape matplotlib ・figure ・pcolormesh ・scatter pytorch ・BCELoss, MSELoss ・device ・Embedding ・TensorDataset, Dataloader ・RNN, LSTM scikit-learn ・SVC ・GaussianNB scipy ・interpolate tkinter ・postscript ・image display ・frame, grid ・Crop Image other ・linear interpolation ・Hysteresis switch ・Square/Triangle wave OpenAI gym ・CartPole-v0 By purpose ・1 of K Coding