win10装anaconda3+cuda+cudnn+tensorflow-gpu+pytorch-gpu最详细教程

说明

说明显卡驱动对应cuda的问题(未来补充)

安装anaconda3

anaconda是一个我见过最好用的python包的管理软件,并且有图形化界面,想装什么包点一下就装了,非常的方便。废话不多说开始装:
1.点击进入anaconda3官网下载页面,下图是官网下载页面截图


2.往下拉,拉到如图所示位置,点击windouw后点击python3.7version下面的Download进行下载

3.下载结束后,双击安装包进行安装,选中图中所选后一路next
4.这就安装好了anaconda,可以在开始菜单里看到有这些选项

安装cuda+cudnn+tensorflow-gpu

传统的cuda和cudnn的安装方法非常的复杂,但是我们现在可以使用anaconda在安装tensorflow-gpu时顺带就把cuda和cudnn安装了,且是适配的,根本不用研究什么哪个tensorflow版本对应那个cuda和cudnn,更不用去了解繁琐cuda和cudnn的安装方法,不得不说ananconda是真的强大。
1.点击图中的Anaconda Prompt,会进入一个命令行
2.在命令行中输入以下命令创建虚拟环境,其中tensorflow_gpu是我们创建的虚拟环境的名字,这个自己可以随意改变的,可以取自己想要的名字。python=3.6代表我们搭建的虚拟环境是用python3.6这个版本建立的,当然你需要使用python3.5或python2.7都是可以自己改的。注意:输入下一行命令后会让我们确认一下,输入y然后回车就好。

conda create -n tensorflow_gpu python=3.6

3.输入以下代码进入我们刚搭建好的虚拟环境,tensorflow_gpu是虚拟环境的名称,应该改写为你自己环境的名称,我这里是tensorflow_gpu:

conda activate tensorflow_gpu

4.输入以下代码安装一些常用的库,这里基本会安装大部分我们将用到的库,一句代码全部安好,anaconda非常强大!

conda install anaconda

5.输入以下代码安装gpu版tensorflow,可以在图中看到,在这一步anaconda就直接给我们安装好了cuda和cudnn,非常的方便。

conda install tensorflow-gpu

5.如果你没有gpu,想安装cpu版本,请输入以下代码(安装gpu版tensorflow请忽略该步):

conda install tensorflow

6.完成以上步骤tensorflow-gpu版(tensorflow-cpu版)就装好了,依次输入以下前两行命令,测试是否装好tensorflow,如果是tensorflow-gpu版本还需要输入最后一行。如果显示如图,那就说明已经装好了tensorflow且能使用gpu计算,可以开始尽情码代码了!!!

python
import tensorflow as tf
sess = tf.Session(config = tf.ConfigProto(log_device_placement=True))

安装pytorch

1.点击进入pytorch官网,往下拉看到如下图所示,前面三行分别选择:stable1.1;windows;conda;后面两行根据当时我们装tensorflow-gpu时装的cuda版本以及python的版本选择,然后输入Run this Command栏中的代码,例如我这里是

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

2.因为我们之前就已经安装好了cuda和cudnn所以pytorch就已经安装好了。接下来可以输入以下代码看pytorch是否安装成功,pytorch是否能使用gpu。如果显示如图,那就说明已经装好了pytorch且能使用gpu计算,可以开始尽情码代码了!!!

python
import torch
torch.cuda.is_available()

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