win10的Pytorch最全安装教程,解决pytorch安装问题!

看着五花八门的pytorch安装教程,作为小白看着实在是眼花缭乱,所以我做一次全攻略安装教程,争取让后来者不走弯路,直接成功安装,不要浪费宝贵的时间。
以下全部是在写博客是的当前版本,如果后续有跟新,可以直接按照类似的方法类比推理安装即可。

1、安装Anaconda

Anaconda至于具体是什么,小白不需要知道,你们大概知道这是一个环境库就完事了,pytorch环境就是在这个里面,所以这个是必须安装的,一下是具体的安装步骤:

  1. 搜索Anaconda;
  2. Geometric Pytorch 安装 pytorch怎么装_CUDA

  3. 直等待,或者找第三方(不推荐,有广告),或者直接你们懂的,毕竟是外网。;
  4. 进入网站后可以看到就是这个样子,直接点击,再选择Download ,接着安装自己对应需要的版本就行。
  5. Geometric Pytorch 安装 pytorch怎么装_人工智能_02


  6. Geometric Pytorch 安装 pytorch怎么装_CUDA_03


  7. Geometric Pytorch 安装 pytorch怎么装_CUDA_04

  8. 接着就是等待下载完成,安装即可,安装我就不说了,既然能看pytorch,我相信安装个软件的能力还是有的,不过你一定要记住自己把Anaconda安装在了什么地方,后期你使用pytorch的时候是需要的;
  9. 安装完成后,在你的开始菜单会有这个,我们主要使用的就是我画圈的那个;
  10. Geometric Pytorch 安装 pytorch怎么装_pytorch_05

2、安装显卡驱动

  1. 首先一定要注意,是显卡驱动,决定CUDA版本,这个一定要记住,我的显卡驱动时466
    ,对应的CUDA版本是11.3,而471之后的版本,对应的是11.4+,但是在我目前写博客的时候,pytorch对应CUDA版本为11.3.所以一定要注意你安装的驱动版本;
  2. 英伟达驱动的话,直接搜索英伟达驱动即可;

    我使用的显卡是3090,填入对应显卡型号,点击开始搜索,点击完事,往下拉页面。会出现下面的页面
    寻找对应的程序安装即可。
  3. 如果不知道自己的驱动是多少,cuda版本是多少,可以打开cmd,输入nvidia-smi即可

3、 安装CUDA

  1. 老样子,还是直接搜cuda,网页打开速度可能也很慢,至于方法,参照Anaconda!

    这里有一个问题,就是点击下载之后,千万不要直接下载,因为下载的是最新版本,我目前最新的是11.5,但是我的驱动是11.3,所以最好是要匹配,要下载历史版本。
  2. 进去之后点击这个
    进入后就可以看到历史版本,因为我是11.3,所以直接找这两个就行

    点击下载,然后安装即可,安装我就不说了,直接无脑下一步就完事了!

4、安装pytorch

  1. 现在是重头戏,终于到了安装pytorch了,首先打开搜索“pytorch官网”,然后点击进入

    可以看到安装有两种方式,一个是pip,一个是conda,我的建议是使用它pip安装,因为我之前使用的conda安装一切就绪,但是跑深度学习就是无法调用cuda,使用的cpu跑的,所以我建议直接pip安装。
  2. 打开anaconda prompt
  3. 输入代码conda create -n pytorch python=3.6,后面的3.6是python的版本,现在我们要创建python的环境,所以根据你自己的设置,我是3.9
  4. 等着全部输入y后,全部done,接着输入conda info --envs
  5. 继续输入代码conda activate pytorch,你可以看到之前代码前面是(base),现在变成了(pytorch),那就说明没问题了。
  6. 接着将这段代码复制进去,变可以安装,仅仅等待即可。
    如果真要是这么安逸就好了,这东西下载极慢,慢的令人发指,也就几k,这个时候,就需要用一下其他方法了。你可以看到最后面其实是一个网址,这个网址就是下载的网址,我们可以复制网址,在浏览器打开。
  7. 打开之后,你可以看到很多东西,你只需要找到自己需要的torch即可!
    给看不懂的小伙伴解释一下,torch-1.10就是torch的版本,cu113是cuda11.3,win是系统,你看上面还有linux的,找到对应的就行。
  8. 用迅雷安排一波,虽然也不是很快,但是要比直接安装快的多!

Geometric Pytorch 安装 pytorch怎么装_CUDA_06


下载完成就是这个,接着上面的第五步,输入代码pip install 你下载文件的绝对地址,完事回车,等他安装就行。

安装完成后千万不要关闭,因为还有两个东西没有安装,接着再把第六步红圈里的代码放进去回车,会自动安装剩下没有安装完成的,这个就非常快了!

  1. 这个时候,你去你的anaconda的文件夹中,可以看到

    需要用到pytorch的话,就直接调用这个。

5、验证cuda是否成功

  1. 进入anaconda prompt中,输入python
  2. 输入import torch,没有报错,再输入torch.cuda.is_available,回ture,那就没啥问题了!或者手头有深度学习的代码,直接跑一下看看是否调用显卡就完事了!

    如果完全按照我的安装,应该不会出现什么问题!
  3. 假如有些同学,想要卸载驱动,重新搞,我建议使用DDU显卡卸载软件,大家直接百度就可以,网上很多,这个可以删除干净,自己很多东西没法删除,比如注册列表啥的。

6、结语

我曾经也是啥也不会,网络上的东西东一句,西一句,看的头脑打晕,所以我写下这篇文章,希望方便后来者,在我的经历中,往往越是小的东西,大佬懒得写,小白不会写,搞得很多人在这种地方浪费了大量的时间,我就是其中一个,希望看到这篇文章的,能够顺利的安装pytorch,快速进入实战环节,不要在安装上浪费宝贵的时间。还有一点,不得不承认,我们的共享精神实在是不如老外,一个小小的东西,都要放到下载中,让花钱下载,我觉得确实不至于,这也是我写这篇文章的另一个起因!