基于CPU的Pytorch安装与测试

Pytorch安装

1. 首先安装anaconda。

2. 在终端的指定conda环境下输入：conda install pytorch torchvision torchaudio cpuonly -c pytorch

Pytorch测试

1. 终端输入python，进入python状态。

2. 输入：

       import torch

       import torchvision

       print(torch.__version__)

       print(torchvision.__version__)

       print(torch.cuda.is_available())

查看torch和torchvision的版本，以及判断cuda是否可用。