週報(大塚)

GPUサーバーのコンテナにPyTorchをインストールする際に少しハマったのでメモを残します。

PyTorchはインストールできたけど、GPUが使えないって人は以下を参考にしてください。

・PyTorch 1.13 からCUDA 10.2 がdeprecatedになったらしい。

PyTorch 1.13 release, including beta versions of functorch and improved support for Apple’s new M1 chips.

・そのため、GET STARTEDからインストールしようとしてもCUDA 10.2が選べない。

・現在のGPUサーバーのCUDAのバージョンは10.2で、そのバージョンのcudatoolkitが入ってないとおそらく動かない。

・コンテナに新しいCUDAをインストールしようとするとコンテナが壊れるみたいなのでやめた方が良い。

今週の進捗(新野)

使用デバイス: cuda:0
-----start-------
Epoch 1/14 | train | Loss: 0.5769 Acc: 0.6843
Epoch 1/14 | val | Loss: 0.4572 Acc: 0.8090
Epoch 2/14 | train | Loss: 0.3569 Acc: 0.8503
Epoch 2/14 | val | Loss: 0.4530 Acc: 0.8256
Epoch 3/14 … Continue Reading ››

Stay Hungry, Stay Foolish!