这篇文章主要介绍python怎么查看pytorch版本,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完!
看代码吧~
import torchprint(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 pandasconda install scikit-learnconda install torchnetexp:安装pyorch2.0 + cuda9.0conda 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.whlpip3 install torchvision
三.检查&查看版本
检查
import torchimport torchvisionprint(torch.cuda.is_available())
查看版本
import torchprint(torch.__version__)
四.可能会遇到的问题:
安装pip3
步骤:
安装setuptools
wget --no-check-certificate https://pypi.python.org/packages/source/s/setuptools/setuptools-19.6.tar.gz#md5=c607dd118eae682c44ed146367a17e26tar -zxvf setuptools-19.6.tar.gzcd setuptools-19.6python3 setup.py buildpython3 setup.py install
安装pip3
wget --no-check-certificate https://pypi.python.org/packages/source/p/pip/pip-8.0.2.tar.gz#md5=3a73c4188f8dbad6a1e6f6d44d117eebtar -zxvf pip-8.0.2.tar.gzcd pip-8.0.2python3 setup.py buildpython3 setup.py install
以上是“python怎么查看pytorch版本”这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注编程网行业资讯频道!