您的位置:首页 > 其它

tensorflow使用gpu进行计算的配置步骤

2017-12-19 18:44 369 查看
1、先安装cudahttps://developer.nvidia.com/cuda-downloads



2、安装cudnn https://developer.nvidia.com/cudnn


3、pip install tensorflow-gpu,如果只需要cpu版本则pip install tensorflow
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: