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

刚换了硬盘,又得重新配置深度学习框架了,现在最流行的莫过于tensorflow和pytorch,我今天就尝试了安装了他们的GPU版本(虽然我的电脑垃圾配置GeForce 930M),以供之后需要的时候参考,也供其他人借鉴
\color{red}{PS:只针对Nvidia显卡可以,具体可看这个,tensorflow有要求}
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卡对应的驱动,然后下载安装,这个过程可能需要耐心等待,其中选项都默认就行

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安装成功

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

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,607评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,047评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,496评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,405评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,400评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,479评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,883评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,535评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,743评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,544评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,612评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,309评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,881评论 3 306
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,891评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,136评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,783评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,316评论 2 342

推荐阅读更多精彩内容