win10+anaconda3安装tensorflow-gpu和pytorch-gpu版

之前在简书写的,但是感觉简书技术氛围不好,所以又转到CSDN中,之后只在简书上发布一些情感文学类博客。

刚换了硬盘,又得重新配置深度学习框架了,现在最流行的莫过于tensorflow和pytorch,我今天就尝试了安装了他们的GPU版本(虽然我的电脑垃圾配置GeForce 930M),以供之后需要的时候参考,也供其他人借鉴
P S : 只 针 对 N v i d i a 显 卡 可 以 , 具 体 可 看 这 个 , t e n s o r f l o w 有 要 求 \color{red}{PS:只针对Nvidia显卡可以,具体可看这个,tensorflow有要求} PSNvidiatensorflow
https://developer.nvidia.com/cuda-gpus

1、需安装的东西

  • Win10 x64位
  • Anaconda3 x64 python 3.6.5
  • CUDA Toolkt10.0
  • cuDNN 7.5 for cuda10
  • Nvidia drive 430.64
  • Visual Studio 2017 community

2、具体实现

2.1 windows10的安装

在我之前写过的安双系统的一篇文章里面有提及安装win10的教程在此不作过多赘述

2.2 Anaconda3的安装(略)

安装过程中提示“不建议添加到环境变量中”,那个默认不勾选就好,要启动到开始里面,到Anaconda Prompt里面用管理员身份打开即可

2.3 Visual Studio 2017 community

现在官网上面已经提供最新的2019下载,之前的版本下载好像要订阅什么的,你们也可以下载我这里提供的链接:https://pan.baidu.com/s/1Zvtg8nHPC_oZBA9rRC_CAA
提取码:s90s ,安装没什么难的,主要之后的显卡驱动要用到VS里面的东西所以要安装它。

2.4 Nvidia drive 430.64的安装

其实我感觉这个是不必须的,因为youtube上面很多教程都没有提及这个,而且我发现后来安装CUDA Toolkit的时候它也会装drive,但是我是装了,你们可以不装试试看
去这里找你电脑的N卡对应的驱动,然后下载安装,这个过程可能需要耐心等待,其中选项都默认就行win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第1张图片
下面是我安好的截图win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第2张图片
安好之后桌面上多了这样一个图标
驱动图标
之后需要重启(在我不重启的时候后面的CUDA10.0安装不成功),至此显卡驱动已安装完成!

2.5 CUDA Toolkt10.0的安装

去这里下载对应版本的,我是win10本地安装,如果不想安装10.1可以在Release Notes里面找10.0版本的安装win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第3张图片这个是安装中的截图:win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第4张图片

win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第5张图片这里面都安装在C盘,默认没有改安装路径,安装成功桌面上出现这样的图标:cuda10.0安装完图标

2.6 cuDNN 7.5 for cuda10的安装

去这里注册下载,这个下载得先注册,选择cuDNN 7.5 for cuda10,因为我们之前装的就是CUDA10.0win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第6张图片下载完就是一个压缩包cudnn.png
解压出来就是一个名为“cuda”的文件夹,里面有win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第7张图片

  • 有些教程里面是说把这三个文件夹里面的.dll文件覆盖掉cuda安装目录下的对应文件,我分析他们之后就没有添加cuda的环境变量了,我没试过
  • 我是按照把C:\cuda\bin添加到环境变量里面了,我认为一样的,我实践证明有效

至此要安装的东西几乎完成了

3、 添加环境变量

除了上面说的把C:\cuda\bin添加到环境变量里面,还要照下图的添加两个环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp

这就是刚刚CUDA的安装目录(当然上面是我自己的啦,你们要看自己的改一下)

4、安装packages

4.1 安装tensorflow-gpu版

进入到Anaconda Prompt里面(因为我没有安装python,这个是anaconda自带的),因为anaconda包挺齐全的,所以直接安装

pip install tensorflow-gpu

我是下载下来.whl文件,然后cd到.whl目录,用

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

这中间我出现过ModuleNotFoundError: No module named ‘numpy.core._multiarray_umath’ 错误,这个只要卸载numpy再重新安装一下就好了

pip uninstall numpy
pip install numpy

下面我测试了一下,看看是否安装成功了
win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第8张图片

4.2 安装pytorch-gpu版

一般都是到官网去用一句话安装,如win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第9张图片但是这上面一般都是最新版的,点击下面的Previous versions of PyTorch,可以找到对应平台的.whl文件下载下来,和上面安装tensorflow的.whl文件一样安装pytorch。win10+anaconda3安装tensorflow-gpu和pytorch-gpu版_第10张图片

至此就实现了win10+anaconda3安装tensorflow-gpu和pytorch-gpu版

如果你觉得有用,就不要吝啬你的喜欢吧,写此教程不易,需要你们的鼓励!!

你可能感兴趣的:(环境搭建代码调试部署,安装类型)