pytorch的安装

pytorch官网：https://pytorch.org/

安装0.4.0版本

pip install torch==0.4.0

pip install torchvision

安装1.0.0版本

pip install torch==1.0.0

pip install torchvision

安装成功的图如下所示：

torch.__version__:查看pytorch 的版本

torch.cuda.is_available():查看cuda是否可用