Google ColabでのPytorchのインストール 環境構築

Question

Google ColabでGPUを使ってPytorchを使用したいのですが、ランタイムをGPUに変更して、

import torch

を実行すると

ValueError                                Traceback (most recent call last)
<ipython-input-36-eb42ca6e4af3> in <module>()
----> 1 import torch

/usr/local/lib/python3.7/dist-packages/torch/__init__.py in <module>()
    195     if USE_GLOBAL_DEPS:
    196         _load_global_deps()
--> 197     from torch._C import *  # noqa: F403
    198 
    199 # Appease the type checker; ordinarily this binding is inserted by the

ValueError: module functions cannot set METH_CLASS or METH_STATIC

というエラーが出できます。ちなみにランタイムをNoneにするとエラーなく実行することができます。解決方法をお願いします

(追加)

!nvidia-smi 
----------------------

Fri Nov 19 04:30:01 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 495.44       Driver Version: 460.32.03    CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Tesla K80           Off  | 00000000:00:04.0 Off |                    0 |
| N/A   60C    P0    59W / 149W |    145MiB / 11441MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

!pip freeze
------------------------------

WARNING: Ignoring invalid distribution -orch (/usr/local/lib/python3.7/dist-packages)
torch==1.7.0+cu101
torchaudio==0.7.0
torchsummary==1.5.1
torchtext==0.11.0
torchvision==0.8.1+cu101
(文字数制限のため省略しました)

Comments

[HRCo4]

今 Colabo にて試してみましたが、こちらは import できました。
torch.__version__ は 1.10.0+cu111 で割り当てられた GPU は Tesla K80 でした。
もう一度試していただいて失敗した場合、割り当てられたGPUとpipの一覧を追記お願いできますか？
!nvidia-smi と !pip freeze で出力できます。

[purogutamu]

上手くできなかったので、追加•編集しました。よろしくお願いします。

[HRCo4]

GPU は同じですが、こちらで起動した環境と異なり大分古いバージョンのようですね。。。
エラー内容で検索をかけると numpy のバージョンが対応していないため発生しているようです。
numpy を一度アンインストールしてバージョン指定でインストールし直してみるといいかもしれません。
torch 1.7.0 に対し、numpy 1.19.4 ではエラーが発生し、1.19.3 ではエラーが表示されなくなったみたいです。

!pip uninstall -y numpy
!pip install numpy==1.19.3

ついでに !nvcc -v で cuda のバージョンも確認しておき、今の torch の cuda (cuda 10.1版)と合っているかの確認もしておいた方がいいかもしれません。

[HRCo4]

参考：
https://discuss.pytorch.org/t/valueerror-and-importerror-occurred-when-import-torch/5818
https://stackoverflow.com/questions/64948852/how-to-solve-the-valueerror-importing-torch-in-python

[purogutamu]

何度もありがとうございます。HRCo4さんが言ったようにnumpyを入れ直したら、torchをimportとすることができました。
感謝しなかないです......本当にありがとうございます。

[purogutamu]

解決できたので、回答してもらえれば、ベストアンサーにしますので、よろしくお願いします。

Answer 1

質問への追記・修正の依頼 と同じになりますが、、、

GPU は同じですが、こちらで起動した環境と異なり大分古いバージョンのようですね。。。
エラー内容で検索をかけると numpy のバージョンが対応していないため発生しているようです。
numpy を一度アンインストールしてバージョン指定でインストールし直してみるといいかもしれません。
torch 1.7.0 に対し、numpy 1.19.4 ではエラーが発生し、1.19.3 ではエラーが表示されなくなったみたいです。

!pip uninstall -y numpy
!pip install numpy==1.19.3

ついでに !nvcc -v で cuda のバージョンも確認しておき、今の torch の cuda (cuda 10.1版)と合っているかの確認もしておいた方がいいかもしれません。