NVIDIA RTX A4000でtensorflowを使いたい

NVIDIA RTX A4000が搭載されたパソコンでtensorflowを使いたいです．
CUDAとかcuDNNとか何に対応しているのかが分かりません．

2021/10/15 10:03

「どこで困っているか」などの詳細を書かないとググる以上の情報は得られません。（大体の場合、どこかでつまずかなければ、ググった内容で済みます）以下も参考にしてください。 https://teratail.com/help/question-tips

siroto

2021/10/16 07:50

迅速な反応ありがとうございます．何とか解決することが出来ました．次からは整理したうえで質問したいと思います．

行動規範の内容に同意します

回答1件

ベストアンサー

NVIDIA RTX A4000は、
NVIDIA RTX A4000 グラフィックスカード
によると「Ampere アーキテクチャ」なので、
Hardware Requirements
の表で「NVIDIA Ampere GPU architecture」と書かれてるところのCUDAとcuDNNのバージョンの組み合わせの中から選ぶ必要があります

また、TensorFlowはバージョンによって、組み合わせて使えるCUDAとcuDNNのバージョンが決まってるので、使いたいTensorFlowのバージョンに合わせてCUDAとcuDNNのバージョンを選ぶ必要があります
おそらく、
ソースからのビルド
や
Windows でのソースからのビルド
の「テスト済みのビルド構成」の表に書かれてるバージョンなら使えると思いますが、どれかのバージョンでは表が間違っていたので、実際入れてみないと正確なところは分かりません
入れてみて使えなかったら、使えるバージョンの組み合わせをググる等して調べて、入れ直すことになります

投稿2021/10/15 13:43