python怎么查看pytorch版本

這篇文章主要介紹python怎么查看pytorch版本，文中介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們一定要看完！

看代碼吧~

import torch
print(torch.__version__)

補充：pytorch不同版本安裝以及版本查看

一：基于conda安裝

conda create --name pytorch_learn python=3.6.7#創(chuàng)建一個名為pytorch_learn的環(huán)境
source activate pytorch_learn #進入環(huán)境
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:
安裝pyorch2.0 + cuda9.0
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

二.基于pip安裝

安裝cuda8.0+torch2.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
print(torch.cuda.is_available())

2.查看版本

import torch
print(torch.__version__)

四.可能會遇到的問題：

安裝pip3

步驟：

第1步：安裝setuptools

wget --no-check-certificate https://pypi.python.org/packages/source/s/setuptools/setuptools-19.6.tar.gz#md5=c607dd118eae682c44ed146367a17e26
tar -zxvf setuptools-19.6.tar.gz
cd setuptools-19.6
python3 setup.py build
python3 setup.py install

第2步：安裝pip3

wget --no-check-certificate https://pypi.python.org/packages/source/p/pip/pip-8.0.2.tar.gz#md5=3a73c4188f8dbad6a1e6f6d44d117eeb
tar -zxvf pip-8.0.2.tar.gz
cd pip-8.0.2
python3 setup.py build
python3 setup.py install

以上是“python怎么查看pytorch版本”這篇文章的所有內容，感謝各位的閱讀！希望分享的內容對大家有幫助，更多相關知識，歡迎關注億速云行業(yè)資訊頻道！