web-dev-qa-db-ja.com

Google Colab GPUにCUDAをインストールする方法

Google Colab GPUにはCUDA Toolkitが付属していないようです。GoogleColab GPUにCUDAをインストールする方法を教えてください。 Google Colabにmxnetをインストールすると、このエラーが発生します。

Installing collected packages: mxnet
Successfully installed mxnet-1.2.0

エラー：計算にGPUを利用するためのインストールが不完全です。 CUDAがインストールされていることを確認して、ターミナルで次の行を実行し、再試行してください。

pip uninstall -y mxnet && pip install mxnet-cu90==1.1.0

CUDAのバージョンに応じて「cu90」を調整します（「cu75」と「cu80」も利用可能です）。 turicreate.config.set_num_gpus（0）を呼び出して、GPUの使用を完全に無効にすることもできます。例外が発生しました。％tbを使用して完全なトレースバックを確認してください。

SystemExit: 1
pythonmachine-learningcudagoogle-colaboratoryturi-create
14
namerbenz

Google ColabにはCudaがプリインストールされていると思います...新しいノートブックを開いて!nvcc --versionこれは、インストールされているCudaバージョンを返します。

ここに私のものがあります： enter image description here

11
Anwarvic
  1. ここに行く： https://developer.nvidia.com/cuda-downloads
  2. Linux-> x86_64-> Ubuntu-> 16.04-> deb（ローカル）を選択します
  3. ダウンロードボタンからリンクをコピーします。
  4. 次に、一連のコマンドを作成する必要があります。最初は、ステップ3で保存したリンクからCUDAインストーラーをダウンロードするwgetの呼び出しです。
  5. 「基本インストーラー」セクションの下にインストール手順があります。それらもコピーしますが、すべての行からSudoを削除します。
  6. 各行の先頭に!のコマンドを追加し、セルに挿入して実行します
  7. 私にとって、コマンドシーケンスは次のとおりです。
    !wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_AMD64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_AMD64.deb !dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_AMD64.deb !apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub !apt-get update !apt-get install cuda
  8. 最後にmxnetをインストールします。上記でインストールしたcudaバージョンは9.2なので、コマンドを少し変更する必要がありました：!pip install mxnet-cu92
  9. Successfully installed graphviz-0.8.3 mxnet-cu92-1.2.0
10
Dienow

GPUの使用に切り替えると、VMでCUDAが利用可能になります。基本的にあなたがする必要があるのは、インストールされているCUDAバージョンとMXNetのバージョンを一致させることです。

以下は、ColabにMXNetをインストールするために使用したものです。

最初にCUDAバージョンを確認します

!cat /usr/local/lib/python3.6/dist-packages/external/local_config_cuda/cuda/cuda/cuda_config.h |\
grep TF_CUDA_VERSION

私にとっては、#define TF_CUDA_VERSION "8.0"

次に、MXNetをインストールしました

!pip install mxnet-cu80
1
Jakub Bartczuk

Colabで実行するには、CUDA 8が必要です（cuda 9+のmxnet 1.1.0は壊れています）。ただし、Google Colabは9.2で実行されます。ただし、9.2をアンインストールし、8.0をインストールしてからmxnet 1.1.0 cu80をインストールする方法があります。

完全なjupyterコードは次のとおりです。 Medium

0
Niko

ここで最も簡単な方法は、mxnet-cu80をインストールすることだと思います。次のコードを使用するだけです。

!pip install mxnet-cu80
import mxnet as mx

また、次の方法で動作するかどうかを確認できます。

a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()

現在colabはcu80をサポートしているだけで、それ以降のバージョンは動作しません。

詳細については、次の2つのWebサイトを参照できます。

Google Colab Free GPU Tutorial

mxnetのインストール

ハッピーコーディング：D

0
Bright Chang