zoukankan      html  css  js  c++  java
  • pytorch安装及基本用法

    20180425更新  安装pytorch0.4.0:

    conda uninstall pytorch 
    # 如果是CUDA版本的话 
    conda uninstall cuda80 cuda90 
    # 如果 已安装的 PyTorch 版本 >= 0.3 
    conda install -c defaults vc=14  
    # 安装官方的包 
    conda install -c pytorch pytorch

    ------------------------分割线--------------------

    1、安装   conda install -c peterjc123 pytorch

    https://zhuanlan.zhihu.com/p/26871672

    后续等官方发布正式版本,安装正式版本。

    1.1CUDA和CUDA使用版本查询

    CUDA:统一计算设备架构技术,是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。

     

    在英伟达官网下载相应显卡版本安装,即可使用。

    版本查询:打开该目录下 version.py即可查询使用版本。   D:Anaconda3pkgspytorch-0.3.0-py36_0.3.0cu90Libsite-packages orch

    2、介绍

    官方0.3.0文档: http://pytorch.org/docs/0.3.0/torch.html

    torch包包含多维张量的数据结构和这些数学类型的运算操作。此外,它提供了有效的序列化张量和任意类型的多种实用工具,还有其他有用的工具。

    它有一个CUDA对等体,使你能够在计算能力> 3的NVIDIA GPU上运行你的张量计算。

    3、用法

    import torch

    dir(torch)

    dir(torch.Tensor)

  • 相关阅读:
    HDU 1087 Super Jumping! Jumping! Jumping!
    HDU 1159 Common Subsequence
    HDU 1003 Maxsum
    HDU 2604 Queuing
    HDU 2045 不容易系列之(3)—— LELE的RPG难题
    HDU 2501 Tiling_easy version
    HDU 2050 折线分割平面
    HDU 2018 母牛的故事
    HDU 2046 骨牌铺方格
    HDU 2044 一只小蜜蜂...
  • 原文地址:https://www.cnblogs.com/bawu/p/8243593.html
Copyright © 2011-2022 走看看