Linux上安装pyTorch
官网安装pyTorch
官网地址:http://pytorch.org/
使用pip安装
进入官网,依次选择OS、Package Manager、Python、CUDA版本,会有对应的安装command命令:
问题
如果出现如下问题:
RuntimeError: module compiled against API version 0xb but this version of numpy is 0xa
则需要更新numpy版本,command:
pip install –user –upgrade numpy