python 、torch、torchvision版本对应关系及安装教程

python 、torch、torchvision版本对应关系及安装教程_第1张图片

pytorch下载连接传送门:https://download.pytorch.org/whl/torch_stable.html

注意:

  1. cpu开头的文件不用我多说,必然是cpu版本的。想下载GPU版本的,找开头是cu的
  2. 举例:cu100/torch-1.2.0-cp36-cp36m-win_amd64.whl,
    从前往后解释:
    ①cu100:表示对应cuda版本为10.0
    ②torch-1.2.0:表示torch版本1.2.0
    ③36:表示适用于python3.6版本
    ④win:window系统(保姆级解释)
    ⑤amd64:对应64位操作系统

下载好对应的版本文件后

  1. 激活需要安装的环境,运行pip install + 文件路径\文件名.whl
  2. 也可以直接打开文件所在文件夹,地址栏输入cmd并回车,直接进入文件所在路径,以我上述介绍的文件为例,一定要先激活已经安装了python3.6.x的环境(因为你下载的是对应python36的torch),再运行pip install +文件名.whl(按tab键可自动搜索文件并补齐)然后回车
  3. 运行pip list查看已安装列表是否有torch,注意不是pytorch
  4. 如果你安装的是GPU版本,检查GPU版torch是否可用:
    命令行依次运行下列命令:
python
import torch
torch.cuda.is_available

如果显示形如下图,则成功:在这里插入图片描述

查看显卡型号

torch.cuda.get_device_name(0)

你可能感兴趣的:(python,pytorch)