pytorch安装

基本环境:Ubuntu14.04.5(64位)+ Python27 + CUDA8.0

Pytorch安装方式
<1> 首推:用conda安装(版本号需要自己给出)

<1.1>  已经安装过CUDA的,根据不同的CUDA版本选择下面的命令

            conda install pytorch=0.1.12 cuda80 -c soumith 

            conda install pytorch=0.1.12 cuda75 -c soumith          

           pytorch=x.x.x 根据自己的需求,表明自己需要的版本号:目前的版本号有:0.1.12, 0.2.0, 0.3.0 等

<1.2>  目前还没有安装CUDA

           conda install pytorch=x.x.x -c soumith    版本号同上,自定义


<2> 其次:用pip安装

<2.1>  下载whl格式安装包,离线安装: pip install /path/to/whl/file

<2.2>  给出whl安装包网络,在线安装:

pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

各种版本的whl的网络路径(下载地址)在这里找到:点这里,需要提醒,有操作系统版本不同,不同Python版本,CUDA支持与否

manylinux1_x86_64(带有manylinux1标识)是不支持CUDA的

<3> 不推荐:源码编译安装,给个链接


主要参考了官方的安装文档

如果不小心被你看到了^_^,欢迎交流你遇到的问题(留言,私信都可以)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值