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来进行计算
时间: 2024-08-28 20:06:48