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)