Fedora31でDeep Learning環境構築
1570 ワード
- 関連ソフトウェアバージョン
- tensorflow-gpu: 2.1
- cuda: 10.1
- cudnn: 7.6.5(cuda 10.1用)
- RPM FusionのHowto/NVIDIAに従って、NVIDIAのドライバを入れる
- RPM FusionのHowto/CUDAに従って、CUDA・CUDNN関係のパッケージを入れる
- 以下の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)
Author And Source
この問題について(Fedora31でDeep Learning環境構築), 我々は、より多くの情報をここで見つけました https://qiita.com/yas99en/items/a824630be811c95be105著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .