• PyTorch图文安装教程(Win10),含遇到的问题及解决办法


    PyTorch安装教程(Win10),含遇到的问题及解决办法

    1. 环境准备

    首先,打开PyTorch官网:https://pytorch.org/get-started/locally/
    向下可以看到有如下几个条件:

    • Python:PyTorch仅支持Python3.x,不支持Python2.x;

    • Package Manager:

      • Anaconda:Anaconda可以去官网下载,在这里贴一个图文教程:https://blog.csdn.net/weixin_43715458/article/details/100096496

      • pip:先到官网上下载pip文件

        1632482888144

        然后将此文件进行解压,打开cmd,将路径调整到解压后的pip文件夹:

        1632483073426

        运行如下命令:

        python setup.py install
        

        1632483205720

        最后为pip添加环境变量,找到python下的scrips,添加至系统变量Path下,记得后面加上分号。(一般在C:\Program Files\Python35\Scripts)

        检查是否安装成功,在cmd中输入pip,返回如下结果:

        1632483791924

    • CUDA (这个很早就安装了,有些忘了)

    2. 安装

    Anaconda安装完成后,点击win菜单里,打开:

    1632485934351

    运行如下指令,创建一个新环境:

    conda create -n pytorch python=3.7
    

    pytorch为环境名称,python=3.7为使用的python版本

    然后会提示是否要安装一些包,输入y即可;

    img

    这样即为新建环境成功,然后输入如下指令,激活环境:

    conda activate pytorch
    

    1632486284224

    这个过程可能会遇到的问题:

    CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/current_repodata.json>
    

    连接错误,我这里的解决办法是:在C盘C:\Users\Administrator打开“.condarc“文件,修改其内容如下:

    ssl_verify: true
    show_channel_urls: true
    
    channels:
     - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
     - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
    

    然后再次打开PyTorch官网,可以在这里选择好指定的版本,打开生成指令中的这个网站:

    pip3 install torch==1.9.1+cu102 torchvision==0.10.1+cu102 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
    

    1632485130413

    打开网站后,根据指令中的版本号选择whl文件,比如我是cu102,torch=1.9.1,以及我使用的是python3.7;

    1632485319173

    然后在pytorch环境、上述下载文件的存储路径下输入:

    pip install torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl
    

    1632481875165

    至此安装成功;

    torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl下载链接:https://pan.baidu.com/s/16mt9FZrMa5UrUqvgqV6MEA
    提取码:pthn

    3. 验证

    验证PyTorch是否安装成功:

    • 进入Python,输入import torch,不报错

    • 输入如下指令,有返回值,表面可以正确使用:

      x = torch.rand(5, 3)
      print(x)
      

      输出类似下图

    • 输入测试gpu和cuda是否可用:

      torch.cuda.is_available()
      
      

      如果可用输出True;

    1632481898622

  • 相关阅读:
    引入css(层叠样式表)文件的三种方法
    pyechart练习二:星巴克门店分布
    搭建Android自动化python+appium环境
    js 数据 删除元素 splice
    SpringCloud: feign整合sentinel实现降级
    电力安全事故安全培训3D仿真展馆绿色环保持久复用
    2.13-CSS基础--清除浮动五种方法
    Java互联网实时聊天系统(附源码)
    Exgcd扩展欧几里得
    为什么创建 Redis 集群时会自动错开主从节点?
  • 原文地址:https://www.cnblogs.com/wwj321/p/15332522.html