看代码吧~
import torch
print(torch.__version__)
补充:pytorch不同版本安装以及版本查看
一:基于conda安装
conda create --name pytorch_learn python=3.6.7#创建一个名为pytorch_learn的环境
source activate pytorch_learn #进入环境
conda install pytorch=0.3.1 cuda80 -c soumith #安装pytorch0.3.1+ cuda8.0(可自己指定)
conda install -c soumith torchvision #安装 torchvision
#安装常用的相关依赖库
conda install pandas
conda install scikit-learn
conda install torchnet
exp:
安装pyorch1.0 + cuda9.0
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
二.基于pip安装
安装cuda8.0+torch1.0+python3.6,可更改链接名指定版本
pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision
三.检查&查看版本
1.检查
import torch
import torchvision