pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装

pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装 

点击进入官网 https://pytorch.org/
方法一:pip安装

Python3.6+pip安装cpu版本

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision


Python3.6+pip安装gpu版本 
目前gpu版本支持cuda8.0,cuda9.0和cuda9.2,请选择对应的版本下载安装,不要同时执行下面三个命令!

pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

  (我选的是这个)
方法二:Conda安装

如果你是Anaconda|Python用户,cpu版本 执行命令:

conda install pytorch-cpu torchvision-cpu -c pytorch


就可以完成安装。

,如果你需要安装cuda8.0、cuda9.0或cuda9.2的gpu版本,请执行:

conda install pytorch torchvision cuda92 -c pytorch

 

 

pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl 


测试安装是否

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值