继续上一篇pytorch安装,这篇记录一下昨天的TensorFlow安装流程。
因为安装pytorch的时候,已经在新机上安装好了CUDA11.0以及cudnn8.0.1,所以在安装TensorFlow的时候不需要再安装了。
直接在命令行中执行:
conda install cudatoolkit=10.1 cudnn=7.6
pip install tensorflow-gpu #安装tensorflow
Note:在安装之前最后换清华源,或者使用豆瓣源。不然会下载超时而失败。
多的不说了,可以参考:
非常详细良心了^_^
工具配置完毕,接下来准备动手了233333