中文官网安装教程:https://www.tensorflow.org/install/install_windows#determine_how_to_install_tensorflow
1.安装前须安装CUDA和cuDNN:
cuDNN需要手动配置的环境变量:
cuDNN:将C:\Program Files\cudnn-9.0-windows10-x64-v7.2.1.38\cuda\bin 添加到path中。
我这里使用的是pip方式安装的:pip3 install --upgrade tensorflow-gpu
注意:win10 的TensorFlow只支持python3版本.使用pip安装时TensorFlow的过程会调用python.exe文件,所以这个版本的python必须是3的,不能是2.7,也不能是调用anaconda中的python3,所以需要在cmd下输入python能调用3这个版本,如下:
2.pip方式安装TensorFlow-gpu 只需要一条命令:pip3 install --upgrade tensorflow-gpu
3.测试是否安装成功
另:anconda方式安装:
前提是输入python后能调用anaconda环境下的python3,就可以按照官网教程安装了。
(完)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Win10+TensorFlow-gpu pip方式安装,anaconda方式安装 - Python技术站