官网安装pyTorch

官网地址:http://pytorch.org/

使用pip安装

进入官网,依次选择OS、Package Manager、Python、CUDA版本,会有对应的安装command命令:
Linux上安装pyTorch

问题

如果出现如下问题:
RuntimeError: module compiled against API version 0xb but this version of numpy is 0xa
则需要更新numpy版本,command:
pip install –user –upgrade numpy

相关文章: