【Pytorch】pytorch安装(包含cuda和cudnn安装)

Pytorch安装

一、安装anaconda

网址:https://www.anaconda.com/products/individual

网页翻到最下面,选择下载Windows 32位/64位版本,傻瓜式安装,记得勾选【Add to Path】 可以将anaconda的环境安装到系统中去

image-20210226184530041

二、Anaconda创建虚拟环境

1 、从开始菜单中打开Anaconda Navigator,单机进入Environments选项卡,如下图所示

image-20210226185001846

2、在Environments选项卡中,单击左下方的Create按钮,创建虚拟环境,如下图所示

image-20210226185132837

image-20210226185258488
此时在Environments选项卡中,已经出现刚刚创建的虚拟环境 【注】才创建的虚拟环境只包含最基本的库

image-20210226185650716

三、安装Pycharm,创建工程

网址:https://www.jetbrains.com/pycharm/

1、单击download进入下载页面

image-20210226185905498

此时,有两种版本选择,嫌麻烦就用社区版,但是功能较少,专业版需要自己找破解,或者用校园邮箱申请一年免费,一年后再次审查通过还可以继续用

image-20210226185939074

2、打开PyCharm,单击New project

image-20210226190033081

3、PyCharm创建Conda工程

image-20210226190145258
image-20210226190227936

4、在Anaconda安装文件夹下找到envs文件夹,里面存储的都是你创建的虚拟环境,然后选择刚刚创建的虚拟环境文件下的python.exe,点击OK

image-20210226190301076
image-20210226190328238
image-20210226190350296

5、创建完成后自动进入如下界面:

image-20210226190429918

四、CUDA安装

1、查看CUDA支持版本

打开NVIDIA控制面板,按照图示顺序查看CUDA所支持的最高版本

image-20210226200016567

2、下载CUDA

打开网址:https://developer.nvidia.com/cuda-toolkit-archive,选择适合自己电脑的cuda版本,比刚才查看的所支持的CUDA 版本低都行

image-20210226233838107

此时要根据自己的系统进行选择

image-20210226235343901
image-20210226235440600

3、下载cuDNN

打开网址:https://developer.nvidia.com/cudnn,点击Download cuDNN

image-20210227000358285

注意,cuDNN下载需要NVIDIA账户,没有则先注册,此步需要耐心,贼卡

image-20210227000437102

4、登陆之后,点击Download cuDNN进入的应该是如下界面,先把问卷填了,才能进入下载界面

image-20210227000708677

此时终于进入下载界面了,勾选I agree…,然后按照下图操作

image-20210227000739017

5、选择适合自己的操作系统版本下载

image-20210227000848436

6、安装CUDA

双击安装包开始安装,此时让你选择的路径并不是安装路径,而是暂时解压出来的文件所需要存放的地方,安装完成后会删除,只要保证选择空间足够大的地方即可,安装路径是默认C盘里,不能修改,此处可以不更改,点击OK即可

image-20210227000933389
image-20210227001012795
image-20210227001038878
image-20210227001105715
image-20210227001140064

验证CUDA是否安装成功,打开cmd,输入 nvcc –V,如下返回则说明安装成功

image-20210227001226296

解压压缩包,压缩包内文件如下:

image-20210227001333119

7、安装cuDNN

打开cuda安装文件夹,默认的强制安装路径为:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\版本号

打开安装路径如下:

image-20210227001422354

将解压的cuDNN压缩包内的三个文件夹复制到CUDA安装目录下:

image-20210227001503720
image-20210227001527682

验证cuDNN是否安装完成,打开cmd,输入

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite

然后执行命令:

bandwidthTest.exe

image-20210227001624983

然后再执行命令:

deviceQuery.exe

image-20210227001657605

8、下载PyTorch安装文件

打开网址:https://download.pytorch.org/whl/torch_stable.html,界面如下:

image-20210227001753290

每个文件命名都是有规律的:

image-20210227001822167

使用快捷键Ctrl+F,搜索我们需要安装的版本文件下载

image-20210227001851897
image-20210227001909013

五、安装pytorch

1、打开PyCharm,进入刚刚创建的工程里,打开Terminal窗口:

image-20210227002109768

2、在PyCharm的Terminal窗口中键入cd 命令,切换目录到PyTorch安装包所在的地址,然后分别执行pip install命令安装两个文件,先安装torch,再安装torchvision,期间会一起下载所需要的依赖包

image-20210227002141655
image-20210227002200869

3、现在验证是否安装成功,在main.py里写入以下代码并运行:

image-20210227002226122
image-20210227002303019

以上方式针对的是windows10操作系统,大家可以根据大家各自的操作系统选择,安装cuda或者不安装cuda

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

推荐阅读更多精彩内容