win7 + tensorflow-gpu + keras + pyTorch安装 环境搭建

软件环境

环境版本
Anaconda3.5.1
visual studiovs2015
cuda9.0
cudnn7.0
python3.6
tensorflow-gpu1.9.0
keras2.2.0
GPUNVIDIA Quadro P2000

Anaconda安装

  1. 在Anaconda的官网上下载需要的版本(Anaconda版本列表)。
  2. 双击运行下载得到.exe文件,默认下一步直至结束。
  3. 配置Annaconda的环境变量,在系统环境变量中添加路径,本文的路径为: D:\software\Anaconda3.5.1\Scripts
  4. 打开cmd命令行进行验证,输入 conda --version 。出现如下窗口,表示anaconda安装成功。conda 版本信息

Visual Stutio安装

安装vs2015以上的版本,网络上有其他教程,不再赘述。
vs下载地址为:https://visualstudio.microsoft.com/zh-hans/vs/

CUDA安装

安装cuda前,检测机器的显卡配置及其对应的驱动程序,根据驱动信息下载对应版本的cuda。

  1. 查看显卡型号:计算机右键–>管理–>设备管理器–>显示适配器
  2. 查看显卡型号对cuda的支持:https://developer.nvidia.com/cuda-gpus
  3. 查看显卡支持的cuda版本:计算机右键–>属性–>控制面板主页–>搜索nvidia–>打开nvidia控制面板–>帮助–> 系统信息–>组件
  4. 下载cuda的本地安装包。
    cuda下载地址:https://developer.nvidia.com/cuda-downloads
    其他版本的cuda地址为:https://developer.nvidia.com/cuda-toolkit-archivecuda下载
  5. 安装cuda前关闭vs2015,双击下载好的.exe文件,选择默认安装直至安装完成。
  6. 配置cuda的环境变量。
变量名
CUDA_PATHC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
CUDA_PATH_V9_0C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
NVCUDASAMPLES_ROOTC:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0
NVTOOLSEXT_PATHC:\Program Files\NVIDIA Corporation\NvToolsExt\
  1. 在cmd命令行中输入 nvcc -V ,出现下图,表示nvcc 编译成功
    cuda版本

cudnn安装

  1. 注册账号,下载对应的cudnn版本 ,下载地址为:https://developer.nvidia.com/rdp/cudnn-archive

cudnn下载
2. 将下载得到的文件cudnn-9.0-windows7-x64-v7.zip进行解压,得到3个子文件夹(bin,lib,include),将这3个子文件夹复制到“NVIDIA GPU Computing Toolkit\CUDA\v9.0”文件夹下(系统会提示已经存在,是否合并,选择确定就可以了),如果找不到文件夹的位置,在C盘搜索CUDA就能找到。
3. 添加环境变量,在path后追加cudnn的bin,本文的环境变量为:D:\cudnn\cuda\bin

pyTorch安装

1.使用conda安装pyTorch时,进入到bash环境或者创建的虚拟环境中
2. 进入pyTorch官网,选择对应的环境和版本信息,网站地址为:https://pytorch.org/
3. 输入conda命令:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
4. 等待安装完成,完成后进入python环境进行测试

import torch
import torchvision
print(torch)

未出现错误信息,则安装完成

tensorflow-gpu安装

  1. 打开Anaconda prompt 创建tensorflow-gpu的虚拟环境
conda create -n tensorflow-gpu-90 python=3.6   
\\在anaconda的 envs文件夹下创建 tensorflow-gpu-90 的虚拟环境,并指定python版本为3.6
  1. 等待环境创建完成,然后进入到该环境中
activate tensorflow-gpu-90   \\ 进入刚创建好的环境中

激活tensorflow-gpu-90虚拟环境
3. 安装指定版本的tensorflow-gpu,也可提前下载对应的tensorflow-gpu的版本。
提供一个阿里云镜像:http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/

pip install tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl

tensorflow-gpu安装
4. 安装完成后进行测试,在该环境中打开python,输入如下代码

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

python
没有出现错误信息,并输出 b'Hello,TensorFlow!'表示安装成功

keras安装

  1. 安装好tensorflow-gpu后,在该环境中使用pip安装keras,并指定keras的版本
pip install keras==2.2

kears安装
2. 等待安装完成,在python环境下进行测试

import tensorflow as tf
import keras

没有输出错误信息,则表示keras安装成功

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值