温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

python怎么查看pytorch版本

发布时间:2021-03-17 09:16:30 来源:亿速云 阅读:2796 作者:小新 栏目:开发技术

这篇文章主要介绍python怎么查看pytorch版本,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完!

看代码吧~

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:
安装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版本”这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注亿速云行业资讯频道!

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI