当前位置 博文首页 > python 如何查看pytorch版本

    python 如何查看pytorch版本

    作者:概率问题 时间:2021-05-09 17:41

    看代码吧~

    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
    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

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持站长博客。如有错误或未考虑完全的地方,望不吝赐教。

    js
    下一篇:没有了