如何找tensorflow-gpu版本对应的cuda和cudnn
生活随笔
收集整理的這篇文章主要介紹了
如何找tensorflow-gpu版本对应的cuda和cudnn
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
第一步
先創建一個虛擬環境(我是在anaconda Prompt中創建):conda create -n TF1.14 python=3.6
第二步
進入虛擬環境conda activate TF1.14
第三步
安裝tensorflow-gpu版本
pip install tensorflow-gpu==1.14.0 -i HTTPS://pypi.doubanio.com/simple/ --trusted host-
(-i后面是加速用的)
第四步
找到創建的虛擬環境路徑:打開D:\anaconda\envs\TF1.14\Lib\site-packages\tensorflow\python\platform下的build_info.py即可,
總結
以上是生活随笔為你收集整理的如何找tensorflow-gpu版本对应的cuda和cudnn的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 在anaconda中安装tensorfl
- 下一篇: 北京大学Tensorflow2.0笔记