tensorflow-GPUのGPU認識に時間がかかる

Question

前提・実現したいこと

tensorflowのGPU認識の時間を早めたい

発生している問題・エラーメッセージ

強化学習時にGPUを使用したいためCUDAなどGPU処理に必要なもの一式を揃えたのですが

device_lib.list_local_devices()

でGPUの認識を行った際

の表示で15分ほど止まった後に

このようにGPUの認識に成功します．

また現在Unityを用いてAnaconda上で強化学習を行っているのですが，学習コードを実行した際も上と同様にGPUの認識に15分ほどかかるため，この認識までにかかる時間を短縮させたいです．

これは設定か何かで短縮できるのでしょうか？それともPCのスペックの問題かそもそもこういう仕様なのでしょうか？

ツールのバージョン，PCのスペック

Anaconda3:2020.11
CUDA:10.1
cudnn:7.6
tensorflow:2.3.1
CPU:AMD Ryzen7 1700
GPU:NVIDA GeForce RTX 3060ti
メモリ:8×2+4×2 24GB

Comments

[jbpb0]

CUDA, cuDNN, ドライバのバージョンを、
https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html
でAmpere architecture(GeForce RTX 30)対応と書かれてるものに合わせたら、もしかしたら直るかもしれません

[tspost]

cuda等を30シリーズに対応したものに変更したら解決しました
ありがとうございます

Answer 1

CUDA：11.2
cudnn：v8.0.5 for CUDA 11.1
tensorflow：2.4.0
numpy:1.19.3
にしたところ瞬間で認識することに成功し無事解決しました
ありがとうございます