質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Windows 10

Windows 10は、マイクロソフト社がリリースしたOSです。Modern UIを標準画面にした8.1から、10では再びデスクトップ主体に戻され、UIも変更されています。PCやスマホ、タブレットなど様々なデバイスに幅広く対応していることが特徴です。

CUDA

CUDAは並列計算プラットフォームであり、Nvidia GPU(Graphics Processing Units)向けのプログラミングモデルです。CUDAは様々なプログラミング言語、ライブラリ、APIを通してNvidiaにインターフェイスを提供します。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

1回答

4360閲覧

NVIDIA RTX A6000でtensorflow-gpu 1.13.1が使えない

shibata-y

総合スコア1

Windows 10

Windows 10は、マイクロソフト社がリリースしたOSです。Modern UIを標準画面にした8.1から、10では再びデスクトップ主体に戻され、UIも変更されています。PCやスマホ、タブレットなど様々なデバイスに幅広く対応していることが特徴です。

CUDA

CUDAは並列計算プラットフォームであり、Nvidia GPU(Graphics Processing Units)向けのプログラミングモデルです。CUDAは様々なプログラミング言語、ライブラリ、APIを通してNvidiaにインターフェイスを提供します。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2021/06/17 08:09

編集2021/06/17 08:22

RTX A6000を使用しておりますが、下記の環境下(tensorflow-gpu 1.13.1)ではエラーが出てしまいます。
一方、tensorflow-gpuをバーション2.1にすると正常に動きます。
tensorflow-gpuバージョンを1.13.1で動かしたいのですが、RTX A6000では使用できないのでしょうか?

エラー内容、症状

  • エラーメッセージにはメモリオーバーフローと記載(たぶん)
  • エラーではないがGPUメモリのロードが異常に遅い

確認内容

 TensorFlowでGPU学習させるためにCUDA周りではまったときの対処法

エラーメッセージ

2021-06-17 13:47:01.520738: E tensorflow/stream_executor/cuda/cuda_blas.cc:698] failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED Traceback (most recent call last): File "shibata-test.py", line 24, in <module> model.fit(x_train, y_train, epochs=5) File "C:\Users\ixs-quadro05\anaconda3\envs\deeplabv3plus\lib\site-packages\tensorflow\python\keras\engine\training.py", line 880, in fit validation_steps=validation_steps) File "C:\Users\ixs-quadro05\anaconda3\envs\deeplabv3plus\lib\site-packages\tensorflow\python\keras\engine\training_arrays.py", line 329, in model_iteration batch_outs = f(ins_batch) File "C:\Users\ixs-quadro05\anaconda3\envs\deeplabv3plus\lib\site-packages\tensorflow\python\keras\backend.py", line 3076, in __call__ run_metadata=self.run_metadata) File "C:\Users\ixs-quadro05\anaconda3\envs\deeplabv3plus\lib\site-packages\tensorflow\python\client\session.py", line 1439, in __call__ run_metadata_ptr) File "C:\Users\ixs-quadro05\anaconda3\envs\deeplabv3plus\lib\site-packages\tensorflow\python\framework\errors_impl.py", line 528, in __exit__ c_api.TF_GetCode(self.status.status)) tensorflow.python.framework.errors_impl.InternalError: Blas GEMM launch failed : a.shape=(32, 512), b.shape=(512, 10), m=32, n=10, k=512 [[{{node dense_1/MatMul}}]] [[{{node loss/mul}}]]

環境

Windows : Windows 10 Pro, バージョン20H2
プロセッサ : Intel(R) Core(TM) i9-10980XE CPU @ 3.00GHz 3.00 GHz
NVIDIAドライバー : R460 U6 (462.31)
--→ファイル名:462.31-quadro-rtx-desktop-notebook-win10-64bit-international-dch-whql.exe
NVIDIA CUDA Toolkit :10.0
--→ファイル名::cuda_10.0.130_411.31_win10.exe
NVIDIA cuDNN:7.4
--→ファイル名:cudnn-10.0-windows10-x64-v7.4.1.5.zip
Anaconda仮想環境内
・python : 3.6.8
・tensorflow-gpu : 1.13.1
*tensorflow-gpuをcondaでインストールするとcudnnなどが自動インストールされるためpipでインストール
(上記でtensorflow-gpu 2.1に変更すると動く!!)

補足

  • tensorflow-gpuバージョンを下記に変更しても動きませんでした
    1.13.1、1.14.0、1.15.0、2.0.0
  • CUDAなどバージョンを下記に変更しても動きません。ただ、tensorflow-gpu 2.1は動きます。
    CUDA Toolkit :10.1、cuDNN:7.6
  • NVIDIAに問い合わせましたが「tensorflowはオープンソースソフトウェアのため回答できません」とのことです

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

ベストアンサー

NVIDIA CUDA Toolkit :10.0

RTX A6000はAmpere architectureです
NVIDIA RTX A6000 グラフィックス カード

Ampere architectureは、現状「CUDA 11.*」のみ対応です
Hardware Requirements

正規に対応してない組み合わせで使ってるのだから、どんな不具合が出てもおかしくないですよね

投稿2021/06/17 10:35

jbpb0

総合スコア7651

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問