Linux上安装pyTorch

来源:互联网 发布:网络剧老炮儿2 编辑:程序博客网 时间:2024/06/03 11:18

官网安装pyTorch

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

使用pip安装

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

问题

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