pycharm+annaconda3+python3.5.2 + 安装tensorflow-gpu版本 [gtx 940mx + Cuda7.0+cudnn v4.0 ]

Posted 2020-10-12

1、安装cuda Toolkit 和cudnn (百度云可下载，版本需要对应)

2、配置环境变量：

3、安装cudnn（需要拷贝一些dll和lib来进行配置）

4、进入cmd，找到anaconda3的pip路径，用下面的命令来执行，可以卸载cpu版本的tensorflow，安装gpu版本的tensorflow

pip uninstall tensorflow
pip install tensorflow-gpu

完成，tensorflow在训练时候会自动调用gpu来进行计算