草庐IT

Pytorch安装,这一篇就够了,绝不踩坑

孔德浩 2023-04-14 原文

在pytorch安装时踩到了不少坑,看了好多博客,最后整合了一份不会踩坑的安装教程,主要是参考各个博主的内容,从零开始安装pytorch,分享给大家!

因为这篇文章是整合各个链接,所以我自己可能写的比较简略,只是为大家提供一个流程,解释的不明白的就点进各位大佬的博客详细看就可以了。

最重要的是:这些链接我会提示你只看指定的位置,不是全部,不是全部,不是全部!

首先在安装pytorch之前,先要安装CUDA,因为一般我们都是用GPU去跑深度学习程序。

cuda安装参考链接:https://blog.csdn.net/Mind_programmonkey/article/details/99688839

安装CUDA时,先看电脑上有没有独立的NVDIA显卡,在设备管理器中查看,如果显卡支持,接下来到回到桌面,右键-NVDIA控制面板-帮助-系统信息,你会看到下图

红色的箭头就是显卡驱动程序版本号,接下来点击组件

这个也就是你最高支持的CUDA版本,目前我这里是11.6,但是我建议大家不需要装最新的,因为目前不容易找到11.6的配套内容,如torch版本等,我有些同学在win11的系统上下载的11.6版本的CUDA,后来发现其他资源不好找,索性又重新换,来回很麻烦,其实低一点点的版本是一样用的,所以我建议还是装前几个版本就够用,下面我用11.3举例

CUDA下载链接:CUDA Toolkit Archive | NVIDIA Developer

点开后点11.3这里,如果下载其他版本的,也是直接点版本就可以

跳转页面如上图选择就可以了,选择后点击下载,下载好后点安装,安装过程大家可以参考我上边分享的CUDA安装的博客,非常详细,安装完以后,接下来就是配置环境变量了,我配置环境变量并没有使用这个up主的方法,而是用的下面这一篇,更详细一些

CUDA环境变量配置

win10下CUDA和CUDNN的安装(超详细)!亲测有效!_没有人喜欢一个人的博客-CSDN博客_cuda安装

下面附上我的环境变量配置

因为我安装了两个版本的CUDA,所以多了一行,大家按11.1样式就可以,这里咱们用11.3举例是一样的,主要是去文件结构下把这些原封不动的粘贴到变量值就可以

然后PATH变量是这样的

我这里太乱了,大家看上文博主规范的就好

注意这个博主箭头这里的目录结构应该和大家不一致,这是他自己创建的文件结构,我建议第一次安装的同学默认c盘,不太能出错

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1

我的路径大概是差不多这样的结构,只要你是默认安装,应该就是这样的,注意一个是Program Files,另一个是ProgramData

到此,cuda就安装好了,记得用博客中的命令行验证安装是否正确,接下来,要安装一个cudnn,参考这个链接,和上文那个是同一篇

win10下pytorch-gpu安装以及CUDA详细安装过程_mind_programmonkey的博客-CSDN博客_pytorch-gpu安装

cudnn下载链接

cuDNN Archive | NVIDIA Developer

点开之后是这样的

我们点击箭头这个

安装windows就可以,安装cudnn需要一个NVDIA账号,按照流程注册一个就好,需要用到邮箱

下载之后,解压缩,将压缩包里面的bincludelib文件直接复制到CUDA的安装目录下,直接覆盖安装。

就是这几个放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1这个目录下,你自己的和我的可能版本号不一样,按你自己的文件结构来就行,覆盖安装

好的,CUDA和cudnn这里就告一段落

然后我们需要下载anaconda,这是一个管理python环境的东西,有了这个,干啥都很方便,这个还是安装非常简单的,我附上一个b站的视频链接吧,跟着大佬一步步做就可以了

Anaconda和pycharm安装教程

【python编程环境安装】全网最详细python环境安装。pycharm和anaconda手把手安装教学。_哔哩哔哩_bilibili

pycharm是一个python编程软件,应该大家都会用到

装好anaconda后,我们来到开始界面,找到这个

点开以后就是命令行,现在我们要建一个Python的虚拟环境来安装pytorch,输入

conda create -n PyTorch python=3.9

这个就是创建环境的语句,这个PyTorch只是一个环境名,自己起就行,建议不要太繁琐,我自己的就叫做kpytorch,然后python版本这里都可以3.8,3.9没什么限制,创建好环境以后,输入activate kpytorch(这个kpytorch就是你的环境名)就到了你的虚拟环境中了,这时输入python就能显示python版本号,进入python编程模式,如果要退出,输入exit()就可以。

接下来一般的安装教程都会让大家去找pytorch的官网,利用网址和镜像源去安装,但是我身边的同学们用这种方式出现了各种各样的错误,所以我建议大家用离线安装:

首先从这个链接下载torch和torchvision

下载链接:https://download.pytorch.org/whl/torch_stable.html

下面这个链接是torch和torchvision的对应关系查询,我下边也附了一张常用图,你的这两个东西的版本和python版本,cuda版本都必须对应好

https://github.com/pytorch/vision#installation

点开下载链接是这样的

往下找,按照cuda11.3举例和python3.9举例,你需要找到

这几个都要对应好,cu113就是cuda11.3,绿色箭头是torch版本,蓝色39就是python版本3.9,然后都有linux和windows两种,选择Windows版本,根据你自己的各个版本去下载对应的whl文件

然后按照torch和torchvision那张图去找你对应的python版本,比如torch1.10.0对应的torchvision版本就是0.11.1

按照上边对应的方法,选择这个就是正确的,下载好两个whl文件后,用anaconda Prompt切换到你的虚拟环境,就是上文那个命令行,在你的虚拟环境中切换到你两个whl文件的安装目录,用cd切换应该都会吧,重点来了:anaconda是不能整体路径切换的,必须要一层一层切换,具体解释看下面这个链接

Anaconda切换盘符不成功:https://blog.csdn.net/c20081052/article/details/88839479

还是用我的举例,注意细看我切路径的方式,我的whl文件就放在D盘的搜狗高速下载中

然后最后一句,在文件名前加pip install去安装这两个文件,是两个文件,都安装完后,就大功告成了

最后验证一下

import torch
torch.cuda.is_available()

这样就OK了

第一次写博客,不知道是否有遗漏的内容,如果大家中途卡在某一步,可以在下方评论

有关Pytorch安装,这一篇就够了,绝不踩坑的更多相关文章

  1. ruby - 在 64 位 Snow Leopard 上使用 rvm、postgres 9.0、ruby 1.9.2-p136 安装 pg gem 时出现问题 - 2

    我想为Heroku构建一个Rails3应用程序。他们使用Postgres作为他们的数据库,所以我通过MacPorts安装了postgres9.0。现在我需要一个postgresgem并且共识是出于性能原因你想要pggem。但是我对我得到的错误感到非常困惑当我尝试在rvm下通过geminstall安装pg时。我已经非常明确地指定了所有postgres目录的位置可以找到但仍然无法完成安装:$envARCHFLAGS='-archx86_64'geminstallpg--\--with-pg-config=/opt/local/var/db/postgresql90/defaultdb/po

  2. ruby - 完全离线安装RVM - 2

    我打算为ruby​​脚本创建一个安装程序,但我希望能够确保机器安装了RVM。有没有一种方法可以完全离线安装RVM并且不引人注目(通过不引人注目,就像创建一个可以做所有事情的脚本而不是要求用户向他们的bash_profile或bashrc添加一些东西)我不是要脚本本身,只是一个关于如何走这条路的快速指针(如果可能的话)。我们还研究了这个很有帮助的问题:RVM-isthereawayforsimpleofflineinstall?但有点误导,因为答案只向我们展示了如何离线在RVM中安装ruby。我们需要能够离线安装RVM本身,并查看脚本https://raw.github.com/wayn

  3. ruby-on-rails - rails 目前在重启后没有安装 - 2

    我有一个奇怪的问题:我在rvm上安装了ruby​​onrails。一切正常,我可以创建项目。但是在我输入“railsnew”时重新启动后,我有“程序'rails'当前未安装。”。SystemUbuntu12.04ruby-v"1.9.3p194"gemlistactionmailer(3.2.5)actionpack(3.2.5)activemodel(3.2.5)activerecord(3.2.5)activeresource(3.2.5)activesupport(3.2.5)arel(3.0.2)builder(3.0.0)bundler(1.1.4)coffee-rails(

  4. ruby - 如何为 emacs 安装 ruby​​-mode - 2

    我刚刚为fedora安装了emacs。我想用emacs编写ruby。为ruby​​提供代码提示、代码完成类型功能所需的工具、扩展是什么? 最佳答案 ruby-mode已经包含在Emacs23之后的版本中。不过,它也可以通过ELPA获得。您可能感兴趣的其他一些事情是集成RVM、feature-mode(Cucumber)、rspec-mode、ruby-electric、inf-ruby、rinari(用于Rails)等。这是我当前用于Ruby开发的Emacs配置:https://github.com/citizen428/emacs

  5. ruby-on-rails - 无法在centos上安装therubyracer(V8和GCC出错) - 2

    我正在尝试在我的centos服务器上安装therubyracer,但遇到了麻烦。$geminstalltherubyracerBuildingnativeextensions.Thiscouldtakeawhile...ERROR:Errorinstallingtherubyracer:ERROR:Failedtobuildgemnativeextension./usr/local/rvm/rubies/ruby-1.9.3-p125/bin/rubyextconf.rbcheckingformain()in-lpthread...yescheckingforv8.h...no***e

  6. ruby - 通过 RVM (OSX Mountain Lion) 安装 Ruby 2.0.0-p247 时遇到问题 - 2

    我的最终目标是安装当前版本的RubyonRails。我在OSXMountainLion上运行。到目前为止,这是我的过程:已安装的RVM$\curl-Lhttps://get.rvm.io|bash-sstable检查已知(我假设已批准)安装$rvmlistknown我看到当前的稳定版本可用[ruby-]2.0.0[-p247]输入命令安装$rvminstall2.0.0-p247注意:我也试过这些安装命令$rvminstallruby-2.0.0-p247$rvminstallruby=2.0.0-p247我很快就无处可去了。结果:$rvminstall2.0.0-p247Search

  7. ruby - 如何在 Lion 上安装 Xcode 4.6,需要用 RVM 升级 ruby - 2

    我实际上是在尝试使用RVM在我的OSX10.7.5上更新ruby,并在输入以下命令后:rvminstallruby我得到了以下回复:Searchingforbinaryrubies,thismighttakesometime.Checkingrequirementsforosx.Installingrequirementsforosx.Updatingsystem.......Errorrunning'requirements_osx_brew_update_systemruby-2.0.0-p247',pleaseread/Users/username/.rvm/log/138121

  8. ruby - Fast-stemmer 安装问题 - 2

    由于fast-stemmer的问题,我很难安装我想要的任何ruby​​gem。我把我得到的错误放在下面。Buildingnativeextensions.Thiscouldtakeawhile...ERROR:Errorinstallingfast-stemmer:ERROR:Failedtobuildgemnativeextension./System/Library/Frameworks/Ruby.framework/Versions/2.0/usr/bin/rubyextconf.rbcreatingMakefilemake"DESTDIR="cleanmake"DESTDIR=

  9. ruby - 安装 Ruby 时遇到问题(无法下载资源 "readline--patch") - 2

    当我尝试安装Ruby时遇到此错误。我试过查看this和this但无济于事➜~brewinstallrubyWarning:YouareusingOSX10.12.Wedonotprovidesupportforthispre-releaseversion.Youmayencounterbuildfailuresorotherbreakages.Pleasecreatepull-requestsinsteadoffilingissues.==>Installingdependenciesforruby:readline,libyaml,makedepend==>Installingrub

  10. ruby - 通过 RVM 安装 Ruby 1.9.2 永远行不通! - 2

    当我执行>rvminstall1.9.2时一切顺利。然后我做>rvmuse1.9.2也很顺利。但是当涉及到ruby​​-v时..sam@sjones:~$rvminstall1.9.2/home/sam/.rvm/rubies/ruby-1.9.2-p136,thismaytakeawhiledependingonyourcpu(s)...ruby-1.9.2-p136-#fetchingruby-1.9.2-p136-#downloadingruby-1.9.2-p136,thismaytakeawhiledependingonyourconnection...%Total%Rece

随机推荐