如何找tensorflow-gpu版本对应的cuda和cudnn

    技术2022-07-11  79

    第一步

    先创建一个虚拟环境(我是在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即可,

    Processed: 0.011, SQL: 9