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:\Anaconda3\pkgs\pytorch-0.3.0-py36_0.3.0cu90\Lib\site-packages\torch

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)