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

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卡对应的驱动,然后下载安装,这个过程可能需要耐心等待,其中选项都默认就行

Nvidia驱动正在安装

下面是我安好的截图

显卡驱动安装成功

安好之后桌面上多了这样一个图标

驱动图标

之后需要重启(在我不重启的时候后面的CUDA10.0安装不成功),至此显卡驱动已安装完成!

 

2.5 CUDA Toolkt10.0的安装

去这里下载对应版本的,我是win10本地安装,如果不想安装10.1可以在Release Notes里面找10.0版本的安装

CUDA Toolkit10.0

这个是安装中的截图:

cuda10.0安装过程图

 

重启后cuda10安装成功

 

这里面都安装在C盘,默认没有改安装路径,安装成功桌面上出现这样的图标:

cuda10.0安装完图标

2.6 cuDNN 7.5 for cuda10的安装

去这里注册下载,这个下载得先注册,选择cuDNN 7.5 for cuda10,因为我们之前装的就是CUDA10.0

cuDNN7.5.1下载

下载完就是一个压缩包

cudnn.png

,解压出来就是一个名为“cuda”的文件夹,里面有

解压出来的cuda中的文件

 

  • 有些教程里面是说把这三个文件夹里面的.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包挺齐全的,所以直接安装

pip3 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

下面我测试了一下,看看是否安装成功了

 

测试tf安装成功

4.2 安装pytorch-gpu版

一般都是到官网去用一句话安装,如

pytorch官网

但是这上面一般都是最新版的,点击下面的Previous versions of PyTorch,可以找到对应平台的.whl文件下载下来,和上面安装tensorflow的.whl文件一样安装pytorch。

pytorch_gpu安装成功

转自简书--赖子NEU
 

你可能感兴趣的:(win10+anaconda3安装tensorflow-gpu和pytorch-gpu版)