Fedora31でDeep Learning環境構築

• 関連ソフトウェアバージョン
  • tensorflow-gpu: 2.1
  • cuda: 10.1
  • cudnn: 7.6.5(cuda 10.1用)
• RPM FusionのHowto/NVIDIAに従って、NVIDIAのドライバを入れる
  • https://rpmfusion.org/Howto/NVIDIA
• RPM FusionのHowto/CUDAに従って、CUDA・CUDNN関係のパッケージを入れる
  • https://rpmfusion.org/Howto/CUDA
• 以下のrpmも入れる

# dnf install libnvinfer-devel-6.0.1-1.cuda10.1 libnvinfer-plugin6-6.0.1-1.cuda10.1

• pipでtensorflow-gpuを入れる

pip install tensorflow-gpu==2.1

• https://keita-blog.com/data_science/keras-tensorflow-gpu の最後に書いてあるサンプルで動作確認

2020-02-11 23:52:11.291003: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1241] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 5201 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1660, pci bus id: 0000:06:00.0, compute capability: 7.5)