2017-05-14 23 views
0

NVIDIA GPUとCUDA 7.5がインストールされたマシンへの非sudoアクセスがあります。私が働いているようだCUDA 7.5のサポート、とPyTorchをインストール:いくつかの練習を取得するにはPyTorch + CUDA 7.5エラー

>>> import torch 
>>> torch.cuda.is_available() 
True 

、私はtutorial for machine translation using RNNsを追いました。 USE_CUDA = Falseを設定してCPUを使用すると、すべて正常に動作します。しかし、ときUSE_CUDA = TrueでのGPUを活用したい、私は次のエラーを取得:私はそのエラーを検索するには、Googleを使用しようとしたが、何の意味のある結果を得なかっまし

Traceback (most recent call last): 
    ... 
    File "seq2seq.py", line 229, in train 
    encoder_output, encoder_hidden = encoder(input_variable[ei], encoder_hidden) 
    File "/.../python2.7/site-packages/torch/nn/modules/module.py", line 206, in __call__ 
    result = self.forward(*input, **kwargs) 
    File "seq2seq.py", line 144, in forward 
    output, hidden = self.gru(embedded, hidden) 
    File "/.../python2.7/site-packages/torch/nn/modules/module.py", line 206, in __call__ 
    result = self.forward(*input, **kwargs) 
    File "/.../python2.7/site-packages/torch/nn/modules/rnn.py", line 91, in forward 
    output, hidden = func(input, self.all_weights, hx) 
    ... 
    File "/.../python2.7/site-packages/torch/backends/cudnn/rnn.py", line 42, in init_rnn_descriptor 
    cudnn.DropoutDescriptor(handle, dropout_p, fn.dropout_seed) 
    File "/usr/lib/python2.7/ctypes/__init__.py", line 383, in __getitem__ 
    func = self._FuncPtr((name_or_ordinal, self)) 
AttributeError: python: undefined symbol: cudnnCreateDropoutDescriptor 
Exception AttributeError: 'python: undefined symbol: cudnnDestroyDropoutDescriptor' in <bound method DropoutDescriptor.__del__ of <torch.backends.cudnn.DropoutDescriptor object at 0x7fe540efec10>> ignored 

を。私はPyTorchとCUDAの初心者なので、ここからどうやって進んでいくのか分かりません。フルセットアップは、Ubuntu 14.04、Python 2.7、CUDA 7.5です。

+1

私が推測するのは、あなたがインストールしたものよりも新しいバージョンのcudnnライブラリを期待しているあなたのリンクです。私はv5またはv5.1より前のバージョンにはその機能がないと思う。 –

+0

@RobertCrovella権利、それはバージョンの問題のようです! CUDA 8.0がインストールされたマシンで実行する機会がありました。 – Christian

答えて

0

コメントに記載されているとおり:エラーは古くなったCUDNNであり、アップグレードすることで解決できます。

現在のバージョンのCUDA、CUDNN、PyTorchをインストールすると、問題ありません。

関連する問題