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文件

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

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

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

      运行如下命令:

      python setup.py install
      

      1632483205720

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

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

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

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

2. 安装

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

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

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

conda create -n pytorch python=3.7

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

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

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

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

conda activate pytorch

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

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

CondaHTTPError: HTTP 000 CONNECTION FAILED for url 

连接错误,我这里的解决办法是:在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

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

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

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

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

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

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

至此安装成功;

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;

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

你可能感兴趣的:(PyTorch图文安装教程(Win10),含遇到的问题及解决办法)