大塚裕介 のすべての投稿

週報(大塚)

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をインストールしようとするとコンテナが壊れるみたいなのでやめた方が良い。