ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

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

2022-02-19 19:02:41  阅读:216  来源: 互联网

标签:python torch pytorch pip 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

至此安装成功;

3. 验证

验证PyTorch是否安装成功:

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

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

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

    输出类似下图

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

    torch.cuda.is_available()
    
    

    如果可用输出True;

1632481898622

标签:python,torch,pytorch,pip,PyTorch,Win10,安装,图文
来源: https://www.cnblogs.com/wwj321/p/15332522.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有