【深度之眼PyTorch框架班三期】pytorch安装 GPU版本(win10,GTX1060,CUDA9.2)

(之前为了安装tensorflow-gpu,看了网上的很多教程,都写的五花八门的。虽然最后是装成功了,但是真的费了很大的劲,花了很多时间。个人的经验是,GPU,CUDA,cudnn还有各种包的版本之间的对应很重要,所以为了安装成功,最好是找一个已经安装成功,并且和你的机器、操作系统、硬件、软件等都非常相似的人写的教程,照着安装,这样成功的几率会比较大(如果有更多的需求可以考虑容器)。这次看了深度之眼余老师的视频,发现他把CUDA,cudnn,pytorch的安装讲的很简单,真的是对我这种小白很友好,所以记录下

首先说明,这里主要是要安装GPU版本的pytorch,还附带了conda虚拟环境安装和pycharm安装。我用的操作系统是win10,安装的CUDA版本是9.2,显卡是GTX1060。

下面就开始吧:(在这之前最好确定已经安装好了显卡的驱动)总体的步骤为:Anaconda安装->pycharm安装->CUDA安装->cudnn安装->pytorch安装(含conda虚拟环境安装,pycharm安装)

一、Anaconda安装

Anaconda下载地址:https://www.anaconda.com/distribution/#download-section

选windows下的python3.7的64-Bit下载。

下载好以后,双击exe文件直接安装。安装过程中一直用它默认的选项就好了,需要注意的是,在下面这一步,这里要注意选择把Anaconda加入到系统环境变量。

然后可以查看anaconda是否安装成功,打开cmd,输入conda,回车。

如果出现类似下面的信息,就是安好了。

二、pycharm安装

去pycharm官网下载→https://www.jetbrains.com/pycharm/download/#section=windows

下载windows的Professional版本。(Professional版要破解,如果不想破解,下载免费的Community版就好了,两者差别很小。)

然后双击exe文件安装,安装过程中一直用它默认的选项就好了,需要注意的是这一步,这里要注意选择把pycharm加入到环境变量。

因为下的是Professional版,要破解,至于怎么破解,也挺简单的,自行百度谷歌下吧,这里就不写了。

三、CUDA安装

因为这里是为了安装pytorch,先去pytorch的官网查看pytorch支持的CUDA版本→https://pytorch.org/get-started/locally/

从下图可以看到,pytorch支持CUDA9.2和10.1,所以这里选择下载CUDA9.2。

CUDA9.2的下载地址→https://developer.nvidia.com/cuda-92-download-archive

点windows,x86_64,10,exe(local),选base installer的download,下载。

双击exe安装

下面检查CUDA是否安装成功

使用cmd进入这个路径(实际上就是CUDA安装在你电脑里面的路径,因人而异)


执行命令: nvcc -V

出现下面类似的结果,就是安装成功了

四、cudnn安装

cudnn下载网址→https://developer.nvidia.com/rdp/cudnn-download。cudnn需要登录账号才可以下载,可以直接注册一个账号。

注册好账号然后登陆,然后就可以下载了。下面开始下载和安装。

勾选下面图中这个

点击如下图所示的地方进行下载。

把这个下载好的cudnn压缩包解压,将解压后里面的三个文件夹

粘贴到之前的CUDA的文件夹下面,如下图所示。

下面检查cudnn是否安装成功

用命令行cmd进入下图路径(下图这个路径在之前说的那个CUDA所在的路径的下面,仔细找找能找到的)

输入命令:bandwidthTest.exe

如果出现pass就表示安装成功

也可以输入命令:deviceQuery.exe

出现了下图的CUDA版本号,GPU型号,还有PASS等信息,就说明CUDA和cudnn安装成功了。

五、pytorch安装(含conda虚拟环境安装,pycharm安装)

先下载pytorch

pytorch下载地址→https://pytorch.org/get-started/locally/

选下图这些

复制这个网址,进入

下载这两个文件,文件名中的cp37代表对应python3.7

下面创建虚拟环境

打开pycharm,点最下面的Terminal

输入这个命令用以创建虚拟环境:conda create -n pytorch_gpu python=3.7

上面的命令中的pytorch_gpu是你给你的虚拟环境取的名字,不一定叫这个名字,可以自己取

在创建的过程中会有一条下图这样的提示,这是虚拟环境保存的地址,要记录,等下会用到。

创建好之后,输入命令conda activate pytorch_gpu进入虚拟环境

出现这个就说明进入虚拟环境了

下面在刚才建立的虚拟环境中安装pytorch

还是在之前那个地方,输入命令 cd XXXXXXXXX
命令中的XXXXXXX表示你自己pytorch的下载后的保存路径

输入 pip install torch,然后按下Tab键

命令自动补全为下图,然后回车安装。

成功后会出现如下提示

输入 pip install torchvision,然后按下Tab键

命令自动补全为下图,然后回车安装。

然后检查pytorch是否安装正确

先新建一个pycharm项目,如何新建可以自行百度

然后把项目和刚刚安好pytorch的虚拟环境关联上,如下图,进入pycharm的设置

如图点击

然后如下图点击

进入之前记录的你虚拟环境保存的地址,点python.exe,此时红框中就会出现类似的地址。然后点OK

等待一会,就关联成功了。

然后检查pytorch是否安装成功

在这个pycharm项目中,新建一个python file,输入如下命令:

import torch

print("pytorch version:{}".format(torch.version)) # 查看pytorch 版本

print(torch.cuda.is_available()) # 查看是否支持GPU

如果出现提示:


说明pytorch安装成功

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

推荐阅读更多精彩内容