網路上找了一堆 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 件のコメント:
コメントを投稿