win10 安装Tensorflow-gpu版本
分类:
文章
•
2024-03-27 08:16:28
亲测在自己电脑上安装Tensorflow,安装成功
第一阶段:安装CUDA和cudnn
- 根据电脑配置去NVIDIA官网下载相应的CUDN版本,我下载的是CUDA9.0
- 根据CUDA版本在NVIDIA官网下载相应的cudnn版本,此处下载的是cudnn v7.5.0
- 执行CUDA的exe文件开始安装
- 安装完之后,解压cudnn压缩包,将解压之后的文件夹中的文件复制到CUDA的相应文件夹中
第二阶段:使用anaconda安装tensorflow-gpu
- 下载Anaconda,我下载的是Anaconda3-2019.03-Windows-x86_64.exe(内置python版本是3.7)
- 安装Anaconda,安装完之后检查Anaconda是否成功安装:conda --version
- 检测目前安装了哪些环境:conda info --envs(目前只显示了基本环境)
- 设置tensorflow环境并安装tensorflow所要用的python版本:conda create --name tensorflow python=3.6(python3.6.8),亲测自带的python3.7不行
- 查看环境conda info --envs,其tensorflow环境已经成功添加
- 切换到tensorflow环境:activate tensorflow(退出当前环境:deactivate)
- 查看tensorflow环境下的python版本:python --version
- 在tensorflow环境下安装GPU版本:pip install tensorflow-gpu==1.12.0(亲测安装1.13.1不成功)
第三阶段:测试tensorflow-gpu是否安装成功
- 进入python命令行:python
- import tensorflow as tf
- 如果步骤2没报错,就差不多了,具体测试见下图