2013-08-10 5 views
7

私はグラフィックカードで動作するようにTheanoをセットアップするのにかなりの手間を掛けています。TheanoにGPUを使用させる

以前はCUDAを使用していましたが、Nvidia Nsightを実行するために必要なように正しくインストールされています。しかし、私は今PyDevでそれを使いたいのですが、チュートリアルの「Using the GPU」の部分に続くいくつかの問題を抱えています。http://deeplearning.net/software/theano/install.html#gpu-linux

最初は非常に基本的なもので、環境変数を設定する方法です。それは私が '$ CUDA_ROOT環境変数'を定義すべきだと言います。いくつかの情報源が、私のホームディレクトリに新しい.pam_environmentファイルを作成すると言っています。 - 謝罪これは基本的な質問であれば、これはまさに、それが書かれなければならな方法であれば

CUDA_ROOT = /usr/local/cuda-5.5/bin 
LD_LIBRARY_PATH = /usr/local/cuda-5.5/lib64/lib 

私はわからない:私は次のようにこれを行われ、書かれています。私はこれが確かにそれを書いた正しい場所であるという確認を得ることができれば、それは助けになるでしょう。

2番目の問題は、チュートリアルの次の部分にあります。 'デバイスオプションを変更して、コンピュータのGPUデバイスの名前を'とします。どうやらこれはTHEANO_FLAGSや.theanorcと関係がありますが、どこにファイルがありますか?もしそうなら、どこで見つけるのですか?チュートリアルは、私が持っていないいくつかの知識を前提としているようです!

これを読んでいただきありがとうございます:すべての回答が大変ありがとうございます - 私は今非常に完全に立ち往生しています!

答えて

1

THEANO_FLAGSは環境変数で、.theanorcは構成ファイルです。両方のメカニズムを使用してTheanoを設定することができます。これはhereと記載されています。

.pam_environmentファイルについて聞いたことがありません。また、あなただけのLD_LIBRARY_PATHの値を上書きしますが、それはこのように/プリペンドを追加しないでください。Theanoについては

LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-5.5/lib64/lib 

、あなたがCUDA_ROOTを定義した場合、あなたはLD_LIBRARY_PATHを変更する必要はありませんので、私はちょうどだろう最後の行を削除します。

通常、あなたのシェルはbashのであれば、人々はこのような.bashrcファイルにENV変数CUDA_ROOTを定義します。

ログアウトして再度ログインした場合の.bashrcへの変更のみが使用されます

export CUDA_ROOT=/usr/local/cuda-5.5/bin 

+0

変更を加えるには、ログアウトする必要はありません。単にソースを実行する〜/ .bashrc – vwvolodya

11
のLinux/OSX上で

編集したり、ファイル~/.theanorcを作成します。ファイルには以下を含める必要があります。

[global] 
floatX = float32 
device = gpu0 

[nvcc] 
fastmath = True 

[cuda] 
root=/usr/local/cuda-5.5/ 
# On a mac, this will probably be /Developer/NVIDIA/CUDA-5.5/ 

$ LD_LIBRARY_PATH変数にcudaを追加する必要があります。 eclipseを実行している場合は、プロジェクトプロパティー>通訳>設定とインタプリタ...>環境に行き、cuda libフォルダ(おそらく/Developer/NVIDIA/CUDA-5.5/lib64)を指すLD_LIBRARY_PATH変数を追加します)

これで、theanoをインポートすると、gpuの検索に関するメッセージが表示されます。 http://deeplearning.net/software/theano/tutorial/using_gpu.htmlにテストコードを実行して、gpuを使用しているかどうかを確認できます。

関連する問題