添加链接
link管理
链接快照平台
  • 输入网页链接,自动生成快照
  • 标签化管理网页链接
Pytorch深度学习入门(准备工作)--Windows10下安装Pytorch(GPU)

Pytorch深度学习入门(准备工作)--Windows10下安装Pytorch(GPU)

最近整理了一下,以前学习Pytorch的学习笔记和相关代码,放在那里也是放着,准备写下来,希望入门Pytorch的同学有帮助!!!

一、安装Anaconda

1. 安装

官网 下载 安装包: anaconda.com/distributi 如果有小伙伴绝得下载的速度慢,可以使用 清华 的镜像: mirrors.tuna.tsinghua.edu.cn

选择你需要的版本是Anaconda3--->python3.6--->64位,下载进行安装,点击I Agree

选择使用永华,点击next,我使用的是默认安装路径,点击next,然后勾选第一项(字体变红也没关系),然后点击Install. 等待安装。

2. 测试安装是否成功

使用win+R,输入cmd打开命令窗口,然后输入conda,若出现一下信息则安装成功:

二、安装pycharm

官网下载: jetbrains.com/pycharm/ 点击download,选择community(免费)下载,下载完成后运行。

点击next,然后选择pycharm安装路径(使用默认也可),然后勾选Add launchers dir to the PATH和.py. 然后点击next,等待安装即可(这个安装过程过于简单,就不放置图片啦)

2023.03.20补充:我目前个人比较喜欢 VScode ,我个人选择 VScode 的原因是,这个编辑器比较小巧一点,我个人比较喜欢里面百度开源插件,会给我们的编程带来不一样的感觉,比如括号,文件标,背景设置等等。其实,我刚开始也是Pycharm,但后面我在德国学习的时候,服务器上只有VScode,所以只能用Vscode,因为服务器安装需要管理员权限,我是没有的。后面使用VScode多了,发现它真的是很方便。

简单说一下Vscode的安装和配置,下载--->安装---->vscode里面下载python就可以。(vscode里面在选择python编辑器也非常方便,右下角就可以)

2023.03.20补充:vscode我比较常用的插件,真心推荐,希望对你有所帮助:

  1. background 它是一个背景设置,本身是一个卡通人物,你可以自己更换你喜欢的图片,比如你女朋友或男朋友的,嘻嘻!

2. Remote-SSH 它是一个远程连接服务器的插件,连接成功后,可以远程修改code,服务器那边自动保存,远程跑程序都可以!

3.Todo Tree 它类似一种标签的功能,你可以设置不同的提醒,比如//TODO这不分程序有问题 //HACK这个不分不确定,非常方便提醒你那部分code没有完成或者有问题

4. Atlantic Night 编辑器主题风格,虽然编程很枯燥,但是有个你自己喜欢的主题,写起来都很开心

5. Better Comments 高亮注释,比较炫酷吧

6. Guides 代码对齐辅助线,很方便,让你阅读起来比较方便

7. indent rainbow 彩虹缩进

8. Rainbow brackets 彩虹括号

9. vscode-icons 文件图标

10 Kite 代码自动补全

三、安装Pytorch_GPU

1.检查是否有GPU,若有,则安装cuda 和cudnn

注意cuda,cudnn,pytorch之间的版本对应关系

2.查看CUDA版本号(Windows)

Step1: 打开控制面板; Step2: 选择查看方式为小图标; Step3: 找出NVIDIA控制面板;

Step4: 选择进入导航栏的帮助中的系统信息; Step5: 点击组件按钮,就可以查看自己电脑的CUDA版本号。

一般采取向下包容原则,安装比自己CUDA版本低的是绝对没有问题的。

3.下载cuda

官网查看: pytorch.org/ 确定对应的cuda版本(这里不用下载)

开始安装, developer.nvidia.com/cu ,选择后下载

(因为我的电脑CUDA的版本是10.0.132,所以这里我选择的是9.2,当然也可以选择和自己电脑相配的别的版本)

下载完成之后运行,我依然选择默认安装路径(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin)

202303.20补充:如果你的电脑是新安装的windows10,默认的安装路径在你安装的时候不是这个,而是C:\Users\xxxx\AppData\Local...,此时也不用担心,你就正常安装,等你安装完成之后,你会发现你的安装路径会变成 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin 这里我简单查询了一下,比较常见的说法是由于windows10在不断的完善,给其他软件的权限是越来越低的,所以默认会变,但是后面后变回来,所以不用担心。

选择精简,下一步安装

验证cuda是否安装成功

进入cuda安装路径,如果你是默认路径,则如下图

复制路径 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin

win+R,cmd打开命令窗,

输入: cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin

输入:nvcc -V

若出现下图信息则安装正确

4.下载cudnn

进入: developer.nvidia.com/rd

选择对应版本的安装包

下载完成后,解压,复制下列三个文件夹

黏贴到cuda安装目录下(如下图)

测试cudnn是否安装成功

进入cuda文件夹找到demo_suite,复制路径:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite

win+R,cmd打开命令窗,输入: cd 黏贴上述路径

回车

输入:bandwidthTest.exe

显示PASS则安装成功

输入:deviceQuery.exe

若显示PASS,则cuda和cudnn均安装成功

5. 安装Pytorch

进入pytorch网站: pytorch.org/ ,选择选择pip下载

可能你的电脑上述选择之后,并不会出现你的CUDA版本,此时也不用担心,点击下面的链接就可以

进入下方网址, download.pytorch.org/wh ,然后你会看见好多版本的一个页面

2023.03.20补充:安装的时候注意选组cuxxx开头的文件,因为我们安装的是gpu版本,我前面安装的是CUDA9.2,因此这里我需要安装cu92开头的文件(如果你是CUDA10.2,就选择cu102,以此类推)。 我选择的是

cu92/torch-1.2.0%2Bcu92-cp36-cp36m-win_amd64.whl

cu92/torchvision-0.4.0%2Bcu92-cp36-cp36m-win_amd64.whl

2023.03.20补充:这里选择是需要注意torch和torchvision版本之间的对应关系,具体可以参考这个 github.com/pytorch/visi ,如下图所示:

2023.03.20补充:这里为什么需要注意torch和torchvision之间的版本关系,是因为,如果你的torch版本是1.8.0,但是你torchvision版本选择了0.8.2,后面安装的时候先安装torch再安装torchvision,当你安装torchvision时它会把你之前的安装的torch1.8.0自动卸载,自动安装1.7.1的cpu版本torch。

注:如果你的python是3.6你就选cp36的,若果是3.7选择cp37,千万不要选择错了!要不版本不匹配!如果绝得下载慢,记得使用国内的镜像

四、配置Pycharm

打开pycharm,新建:file->new project,写个工程名pytorch_learning;

工程名右击,新建:new->python file->test_pytorch.py

输入

import torch
print(torch.__version__)

然后,你会出现错误,这很正常,因为pytorch还没有安装完,记得你刚刚下的两个东西还没用上。

莫急,接下来,我们在Anaconda下创建一个环境,这样做事非常有好处的,具体好处就不提啦!你可以问问度娘!

具体步骤:打开pycharm下的terminal,然后出入 conda create -n pytorch_gpu python=3.6 (pytorch_gpu是新建环境的名字,也可写成其他任意的)

如果安装不成功,应该是显示升级conda命令,按照它给定提示升级就可以

激活环境:conda activate pytorch_gpu

2023.03.20补充:关闭环境 conda deactivate 删除环境 conda remove -n 需要删除的环境名 --all

查看环境列表 conda env list

五、创建的环境中安装pytorch

复制你下载

cu92/torch-1.2.0%2Bcu92-cp36-cp36m-win_amd64.whl

cu92/torchvision-0.4.0%2Bcu92-cp36-cp36m-win_amd64.whl

这两个东西的路径path,然后在pycharm的terminal中 cd path, 然后使用如下命令进行安装:

pip install “torch-1.2.0+cu92-cp36-cp36m-win_amd64.whl”

pip install "torchvision-0.4.0+cu92-cp36-cp36m-win_amd64.whl"

这里基本不会出现错误!( 2023.03.20补充:这里可能会出现安装其它的库,导致安装超时,此时可以在上述命令后面添加-i pypi.tuna.tsinghua.edu.cn (这个是清华的镜像,当然也可以选择其它镜像,我个人比较喜欢这个),具体为:pip install torchvision-0.40+cu92-cp36-cp36m-win_amd64.whl -i pypi.tuna.tsinghua.edu.cn

六、将虚拟环境pytorch_gpu关联到项目中

打开pycharm,找到导航栏的file->settings->

然后点那个下三角,然后show all,在点绿色的加号,然后点击Add Local,找到你的Anaconda下的python解释器就可以。


然后,再次运行刚刚的test_pytorch.py文件,出现

也可以使用print(torch.cuda.is_available())进项验证,输出True,表示安装成功

恭喜你,你成功啦!赶紧用Pytorch写几个小程序跑一跑,感受一下它的魅力吧!

后续我会更新pytorch的相关知识和操作,草稿箱已经写好了标题!刚刚看了一下一共是53篇,不对52篇,去掉这一篇!有时间我就会更新,不出意外应该是每周的周六或周日!欢迎大家和我一起讨论学习!


参考博客和书籍:

1. blog.csdn.net/qq_390224

2. jianshu.com/p/d3b9419a0

3. blog.csdn.net/huima2017

4. blog.csdn.net/kan228112

4.Pytorch深度学习入门(曾芃壹)

编辑于 2023-03-20 18:02 ・IP 属地吉林

文章被以下专栏收录