cuda=10.0 安装 pytorch=1.4.0+torchvision=0.5.0 或 cpu安装pytorch

pytorch+torchvision 安装

---

不同的安装命令可到PyTorch官网查看：PyTorch官网
点击Get-Started，到如下界面查找适合自己配置的命令。

下面分别介绍GPU方式安装和CPU方式安装

---

GPU方式安装：

首先用如下命令查看CUDA版本

cat /usr/local/cuda/version.txt

本机CUDA 版本为10.0
再安装 pytorch 和 torchvision

 pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

安装成功！

---

CPU方式安装：

直接采用命令方式安装一直没成功，最后采用whl方式安装，成功。
下载地址：https://download.pytorch.org/whl/torch_stable.html
ctrl-F查找，找到 torch=1.4，cpu版本，操作系统版本为windows，以及适合自己python版本的whl文件并下载。（cp37表示适合python3.7版本）

我下载的是 cpu/torch-1.4.0%2Bcpu-cp37-cp37m-win_amd64.whl
下载后的文件名为：torch-1.4.0+cpu-cp37-cp37m-win_amd64.whl
cd 到该.whl文件所在的文件夹，利用以下命令即可安装。

pip install torch-1.4.0+cpu-cp37-cp37m-win_amd64.whl

再安装 torchvision

pip install torchvision ==0.5.0

简单测试一下是否安装成功，运行以下代码检验：

import torch
import torchvision
print(torch.__version__)

能正确输出信息则安装成功！

推荐博客：torch 与 torchvision 版本对应关系

Good Luck！