2019年5月27日月曜日

[Deep Learning] 用 RTX2070 跑 tensorflow-gpu


網路上找了一堆 solution 結果都沒有能用的
最後找到了一個可以成功的
所以記錄一下

原本用 CPU 跑都沒事,但是換成 GPU 之後就一直出現這個錯誤
看了半天感覺是版本的問題
試了半天結果都沒什麼效果
















後來找到了下面這篇
[RTX2080でTensorflow-gpu]
http://blog.abars.biz/archives/52458625.html

裡面有提到
去下面這個地方抓人家打包好的檔案來用
https://github.com/fo40225/tensorflow-windows-wheel

CUDA Toolkit:
RTX 20 系列基本上可以用 cuda v10 以上的版本

cuDNN:
根據上面選的 whl 檔來決定要裝哪個版本就好
這個相對沒影響

0 件のコメント:

コメントを投稿