ubuntu 安装 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)
这里使用pip安装pytorch,我试过cuda安装,和电脑的配置没兼容,没有安装成功,后来发现使用pip安装很简单方便,就是用pip安装
首先进入pytorch官网,往下拉会看到不同的安装选项
根据自己电脑的配置选择不同的选项,例如我选择的stable版本、Linux系统、pip安装、python版本为2.7、cuda为8.0,如下图所示:
选择完之后下面会出现安装所需的命令,如下图
先后运...
原创
2018-10-23 00:15:19 ·
19610 阅读 ·
0 评论